lecrapaud 0.16.2__tar.gz → 0.16.4__tar.gz

{lecrapaud-0.16.2 → lecrapaud-0.16.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: lecrapaud
-Version: 0.16.2
+Version: 0.16.4
 Summary: Framework for machine and deep learning, with regression, classification and time series analysis
 License: Apache License
 Author: Pierre H. Gallet

{lecrapaud-0.16.2 → lecrapaud-0.16.4}/lecrapaud/db/alembic.ini

@@ -82,7 +82,7 @@ sqlalchemy.url = %(DATABASE_URL)s
 
 # Logging configuration
 [loggers]
-keys = root,sqlalchemy,alembic
+keys = root,sqlalchemy,alembic,lecrapaud
 
 [loggers_root]
 disable_existing_loggers = False
@@ -108,6 +108,12 @@ level = INFO
 handlers =
 qualname = alembic
 
+[logger_lecrapaud]
+level = INFO
+handlers = console
+qualname = lecrapaud
+propagate = 0
+
 [handler_console]
 class = StreamHandler
 args = (sys.stderr,)

{lecrapaud-0.16.2 → lecrapaud-0.16.4}/lecrapaud/model_selection.py

@@ -883,7 +883,7 @@ class ModelSelectionEngine:
                         model_best_params = json_dict[model_name]
                 except Exception:
                     raise FileNotFoundError(
-                        f"Could not find {model_name} in current data. Try to run an hyperoptimization by setting `perform_hyperopt` to true, pass `best_params`"
+                        f"Could not find {model_name} in current data. Try to run an hyperoptimization by setting `perform_hyperopt` to true, or pass `best_params`"
                     )
 
             # save best params
@@ -1133,6 +1133,27 @@ class ModelSelectionEngine:
 
         storage_path = f"{self.results_dir}/ray_results"
 
+        # Initialize Ray with the runtime environment
+        ray.init(
+            runtime_env={
+                "excludes": [
+                    ".git/**/*",
+                    "**/*.pyc",
+                    "**/__pycache__",
+                    "**/data/*",
+                    "**/notebooks/*",
+                    "**/tests/*",
+                    "**/docs/*",
+                    "**/.pytest_cache/*",
+                    "**/venv/*",
+                    "**/.venv/*",
+                    "**/build/*",
+                    "**/dist/*",
+                    "**/*.egg-info/*",
+                ]
+            }
+        )
+
         tuner = Tuner(
             trainable=with_parameters(
                 trainable,
@@ -1162,23 +1183,6 @@ class ModelSelectionEngine:
                 stop={"training_iteration": 100},
                 storage_path=storage_path,
                 callbacks=[TBXLoggerCallback()],
-                runtime_env={
-                    "excludes": [
-                        ".git/**/*",
-                        "**/*.pyc",
-                        "**/__pycache__",
-                        "**/data/*",
-                        "**/notebooks/*",
-                        "**/tests/*",
-                        "**/docs/*",
-                        "**/.pytest_cache/*",
-                        "**/venv/*",
-                        "**/.venv/*",
-                        "**/build/*",
-                        "**/dist/*",
-                        "**/*.egg-info/*",
-                    ]
-                },
             ),
         )
         try:
@@ -1249,7 +1253,7 @@ class ModelSelectionEngine:
                 try:
                     with open(best_params_file, "r") as f:
                         target_params = json.load(f)
-                        all_best_params[target_number] = target_params
+                        all_best_params[int(target_number)] = target_params
                 except Exception as e:
                     logger.warning(
                         f"Error loading best params for {target_dir}: {str(e)}"

{lecrapaud-0.16.2 → lecrapaud-0.16.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lecrapaud"
-version = "0.16.2"
+version = "0.16.4"
 description = "Framework for machine and deep learning, with regression, classification and time series analysis"
 authors = [
     {name = "Pierre H. Gallet"}