wavetrainer 0.0.10__tar.gz → 0.0.11__tar.gz

{wavetrainer-0.0.10/wavetrainer.egg-info → wavetrainer-0.0.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wavetrainer
-Version: 0.0.10
+Version: 0.0.11
 Summary: A library for automatically finding the optimal model within feature and hyperparameter space.
 Home-page: https://github.com/8W9aG/wavetrainer
 Author: Will Sackfield

{wavetrainer-0.0.10 → wavetrainer-0.0.11}/setup.py

@@ -23,7 +23,7 @@ def install_requires() -> typing.List[str]:
 
 setup(
     name='wavetrainer',
-    version='0.0.10',
+    version='0.0.11',
     description='A library for automatically finding the optimal model within feature and hyperparameter space.',
     long_description=long_description,
     long_description_content_type='text/markdown',

{wavetrainer-0.0.10 → wavetrainer-0.0.11}/wavetrainer/__init__.py

@@ -2,5 +2,5 @@
 
 from .create import create
 
-__VERSION__ = "0.0.10"
+__VERSION__ = "0.0.11"
 __all__ = ("create",)

{wavetrainer-0.0.10 → wavetrainer-0.0.11}/wavetrainer/create.py

@@ -1,5 +1,7 @@
 """A function for creating a new trainer."""
 
+# pylint: disable=too-many-arguments,too-many-positional-arguments
+
 import datetime
 
 from .trainer import Trainer
@@ -11,6 +13,7 @@ def create(
     test_size: float | datetime.timedelta | None = None,
     validation_size: float | datetime.timedelta | None = None,
     dt_column: str | None = None,
+    max_train_timeout: datetime.timedelta | None = None,
 ) -> Trainer:
     """Create a trainer."""
     return Trainer(
@@ -19,4 +22,5 @@ def create(
         test_size=test_size,
         validation_size=validation_size,
         dt_column=dt_column,
+        max_train_timeout=max_train_timeout,
     )

{wavetrainer-0.0.10 → wavetrainer-0.0.11}/wavetrainer/trainer.py

@@ -51,6 +51,7 @@ class Trainer(Fit):
         test_size: float | datetime.timedelta | None = None,
         validation_size: float | datetime.timedelta | None = None,
         dt_column: str | None = None,
+        max_train_timeout: datetime.timedelta | None = None,
     ):
         tqdm.tqdm.pandas()
 
@@ -137,6 +138,7 @@ class Trainer(Fit):
         self._test_size = test_size
         self._validation_size = validation_size
         self._dt_column = dt_column
+        self._max_train_timeout = max_train_timeout
 
     def _provide_study(self, column: str) -> optuna.Study:
         storage_name = f"sqlite:///{self._folder}/{column}/{_STUDYDB_FILENAME}"
@@ -289,7 +291,12 @@ class Trainer(Fit):
                 initial_trials = max(self._trials - len(study.trials), 0)
             if initial_trials > 0:
                 study.optimize(
-                    test_objective, n_trials=initial_trials, show_progress_bar=True
+                    test_objective,
+                    n_trials=initial_trials,
+                    show_progress_bar=True,
+                    timeout=None
+                    if self._max_train_timeout is None
+                    else self._max_train_timeout.total_seconds(),
                 )
 
             train_len = len(df[dt_index < start_test_index])
@@ -336,6 +343,9 @@ class Trainer(Fit):
                             validate_objctive, idx=test_idx, series=test_series
                         ),
                         n_trials=1,
+                        timeout=None
+                        if self._max_train_timeout is None
+                        else self._max_train_timeout.total_seconds(),
                     )
 
                 _fit(study.best_trial, test_df, test_series, True, test_idx)

{wavetrainer-0.0.10 → wavetrainer-0.0.11/wavetrainer.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wavetrainer
-Version: 0.0.10
+Version: 0.0.11
 Summary: A library for automatically finding the optimal model within feature and hyperparameter space.
 Home-page: https://github.com/8W9aG/wavetrainer
 Author: Will Sackfield