mostlyai-engine 1.7.0__tar.gz → 1.7.1__tar.gz

{mostlyai_engine-1.7.0 → mostlyai_engine-1.7.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mostlyai-engine
-Version: 1.7.0
+Version: 1.7.1
 Summary: Synthetic Data Engine
 Project-URL: homepage, https://github.com/mostly-ai/mostlyai-engine
 Project-URL: repository, https://github.com/mostly-ai/mostlyai-engine

{mostlyai_engine-1.7.0 → mostlyai_engine-1.7.1}/mostlyai/engine/__init__.py

@@ -22,7 +22,7 @@ from mostlyai.engine.splitting import split
 from mostlyai.engine.training import train
 
 __all__ = ["split", "analyze", "encode", "train", "generate", "init_logging", "set_random_state"]
-__version__ = "1.7.0"
+__version__ = "1.7.1"
 
 # suppress specific warning related to os.fork() in multi-threaded processes
 warnings.filterwarnings("ignore", category=DeprecationWarning, message=".*multi-threaded.*fork.*")

{mostlyai_engine-1.7.0 → mostlyai_engine-1.7.1}/mostlyai/engine/_language/training.py

@@ -15,7 +15,6 @@
 import gc
 import json
 import logging
-import math
 import time
 import warnings
 from collections.abc import Callable
@@ -355,12 +354,7 @@ def train(
         max_training_time = max(0.0, max_training_time * 60)  # convert to seconds
         _LOG.info(f"{max_training_time=}s")
         max_epochs = max(0.0, max_epochs)
-        max_epochs_cap = math.ceil((trn_cnt + val_cnt) / 25)
-        if max_epochs_cap < max_epochs:
-            _LOG.info(f"{max_epochs=} -> max_epochs={max_epochs_cap} due to small sample size")
-            max_epochs = max_epochs_cap
-        else:
-            _LOG.info(f"{max_epochs=}")
+        _LOG.info(f"{max_epochs=}")
 
         _LOG.info(f"{with_dp=}")
         _LOG.info(f"{model_state_strategy=}")

{mostlyai_engine-1.7.0 → mostlyai_engine-1.7.1}/mostlyai/engine/_tabular/training.py

@@ -13,7 +13,6 @@
 # limitations under the License.
 
 import logging
-import math
 import time
 import warnings
 from collections.abc import Callable
@@ -391,12 +390,7 @@ def train(
         max_training_time = max(0.0, max_training_time) * 60  # convert to seconds
         _LOG.info(f"{max_training_time=}s")
         max_epochs = max(0.0, max_epochs)
-        max_epochs_cap = math.ceil((trn_cnt + val_cnt) / 25)
-        if max_epochs_cap < max_epochs:
-            _LOG.info(f"{max_epochs=} -> max_epochs={max_epochs_cap} due to small sample size")
-            max_epochs = max_epochs_cap
-        else:
-            _LOG.info(f"{max_epochs=}")
+        _LOG.info(f"{max_epochs=}")
         model_sizes = {
             "MOSTLY_AI/Small": ModelSize.S,
             "MOSTLY_AI/Medium": ModelSize.M,

{mostlyai_engine-1.7.0 → mostlyai_engine-1.7.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mostlyai-engine"
-version = "1.7.0"
+version = "1.7.1"
 description = "Synthetic Data Engine"
 authors = [{ name = "MOSTLY AI", email = "dev@mostly.ai" }]
 requires-python = ">=3.10"