lecrapaud 0.16.2__py3-none-any.whl → 0.16.4__py3-none-any.whl

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/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.dist-info → lecrapaud-0.16.4.dist-info}/METADATA

@@ -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.dist-info → lecrapaud-0.16.4.dist-info}/RECORD

@@ -9,7 +9,7 @@ lecrapaud/db/alembic/versions/2025_06_23_1748-f089dfb7e3ba_.py,sha256=hyPW0Mt_B4
 lecrapaud/db/alembic/versions/2025_06_24_1216-c62251b129ed_.py,sha256=6Pf36HAXEVrVlnrohAe2O7gVaXpDiv3LLIP_EEgTyA0,917
 lecrapaud/db/alembic/versions/2025_06_24_1711-86457e2f333f_.py,sha256=KjwjYvFaNqYmBLTYel8As37fyaBtNVWTqN_3M7y_2eI,1357
 lecrapaud/db/alembic/versions/2025_06_25_1759-72aa496ca65b_.py,sha256=MiqooJuZ1etExl2he3MniaEv8G0LrmqY-0m22m9xKmc,943
-lecrapaud/db/alembic.ini,sha256=TXrZB4pWVLn2EUg867yp6paA_19vGeirO95mTPA3nbs,3699
+lecrapaud/db/alembic.ini,sha256=Zw2rdwsKV6c7J1SPtoFIPDX08_oTP3MuUKnNxBDiY8I,3796
 lecrapaud/db/models/__init__.py,sha256=Lhyw9fVLdom0Fc6yIP-ip8FjkU1EwVwjae5q2VM815Q,740
 lecrapaud/db/models/base.py,sha256=J9ew-0z_-tnWAwhVvOmVDys2R6jPF_oSca_ny6wpXQE,7606
 lecrapaud/db/models/experiment.py,sha256=mQmbp_LzwMYVLzwYlaoRWYgzZCSJCA9dDcLpuvddsms,10931
@@ -36,10 +36,10 @@ lecrapaud/misc/tabpfn_tests.ipynb,sha256=VkgsCUJ30d8jaL2VaWtQAgb8ngHPNtPgnXLs7QQ
 lecrapaud/misc/test-gpu-bilstm.ipynb,sha256=4nLuZRJVe2kn6kEmauhRiz5wkWT9AVrYhI9CEk_dYUY,9608
 lecrapaud/misc/test-gpu-resnet.ipynb,sha256=27Vu7nYwujYeh3fOxBNCnKJn3MXNPKZU-U8oDDUbymg,4944
 lecrapaud/misc/test-gpu-transformers.ipynb,sha256=k6MBSs_Um1h4PykvE-LTBcdpbWLbIFST_xl_AFW2jgI,8444
-lecrapaud/model_selection.py,sha256=dz9gRLUEkaOFo8EneS6t7GJRSJXKyWmbbYRRulIKE1Q,68492
+lecrapaud/model_selection.py,sha256=tjBM9X5CP02pNiuNLdNK2MFZblrH7w-gQgBc5FEN-TY,68514
 lecrapaud/search_space.py,sha256=-JkzuMhaomdwiWi4HvVQY5hiw3-oREemJA16tbwEIp4,34854
 lecrapaud/utils.py,sha256=JdBB1NvbNIx4y0Una-kSZdo1_ZEocc5hwyYFIZKHmGg,8305
-lecrapaud-0.16.2.dist-info/LICENSE,sha256=MImCryu0AnqhJE_uAZD-PIDKXDKb8sT7v0i1NOYeHTM,11350
-lecrapaud-0.16.2.dist-info/METADATA,sha256=TPKjiK_wQekvDhTXf8XXg_TtkIRdizJoJcawSmdrfQY,11081
-lecrapaud-0.16.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-lecrapaud-0.16.2.dist-info/RECORD,,
+lecrapaud-0.16.4.dist-info/LICENSE,sha256=MImCryu0AnqhJE_uAZD-PIDKXDKb8sT7v0i1NOYeHTM,11350
+lecrapaud-0.16.4.dist-info/METADATA,sha256=cJJFa99bkNUUq21Tyn1ykT9vMzqMXzQ-iVgcm-mcM3k,11081
+lecrapaud-0.16.4.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+lecrapaud-0.16.4.dist-info/RECORD,,