Merge pull request #16 from MAIF/fix/catboost_writing_files

githubyako · web-flow · commit ec9779738e22 · 2022-05-09T18:06:43.000+02:00
Fix/catboost writing files
diff --git a/eurybia/__version__.py b/eurybia/__version__.py
@@ -1,3 +1,3 @@
-VERSION = (1, 0, 0)
+VERSION = (1, 0, 1)
 
 __version__ = ".".join(map(str, VERSION))
diff --git a/eurybia/core/smartdrift.py b/eurybia/core/smartdrift.py
@@ -290,6 +290,7 @@ def compile(
             loss_function=hyperparameter["loss_function"],
             eval_metric=hyperparameter["eval_metric"],
             task_type="CPU",
+            allow_writing_files=False
         )
 
         datadrift_classifier = datadrift_classifier.fit(train_pool_cat, eval_set=test_pool_cat, silent=True)

Original file line number	Diff line number	Diff line change
`@@ -290,6 +290,7 @@ def compile(`
`290`	`290`	`loss_function=hyperparameter["loss_function"],`
`291`	`291`	`eval_metric=hyperparameter["eval_metric"],`
`292`	`292`	`task_type="CPU",`
	`293`	`+ allow_writing_files=False`
`293`	`294`	`)`
`294`	`295`
`295`	`296`	`datadrift_classifier = datadrift_classifier.fit(train_pool_cat, eval_set=test_pool_cat, silent=True)`