mostlyai-engine 1.1.3__tar.gz → 1.1.4__tar.gz

{mostlyai_engine-1.1.3 → mostlyai_engine-1.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mostlyai-engine
-Version: 1.1.3
+Version: 1.1.4
 Summary: Synthetic Data Engine
 Project-URL: homepage, https://github.com/mostly-ai/mostlyai-engine
 Project-URL: repository, https://github.com/mostly-ai/mostlyai-engine
@@ -24,15 +24,16 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Software Development :: Libraries
 Classifier: Typing :: Typed
 Requires-Python: >=3.10
-Requires-Dist: accelerate<0.32,>=0.31.0
-Requires-Dist: datasets<3,>=2.20.0
+Requires-Dist: accelerate>=0.31.0
+Requires-Dist: datasets<4,>=3.0.0
 Requires-Dist: formatron<0.5,>=0.4.11
+Requires-Dist: huggingface-hub<0.29,>=0.28.0
 Requires-Dist: joblib<2,>=1.3.0
-Requires-Dist: json-repair<0.31,>=0.30.0
+Requires-Dist: json-repair>=0.30.0
 Requires-Dist: numpy<2,>=1.26.3
 Requires-Dist: opacus<2,>=1.5.2
 Requires-Dist: pandas~=2.2.0
-Requires-Dist: peft<0.12,>=0.11.1
+Requires-Dist: peft>=0.12.0
 Requires-Dist: psutil<6,>=5.9.5
 Requires-Dist: pyarrow>=16.0.0
 Requires-Dist: rich<14,>=13.9.4

{mostlyai_engine-1.1.3 → mostlyai_engine-1.1.4}/mostlyai/engine/__init__.py

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

{mostlyai_engine-1.1.3 → mostlyai_engine-1.1.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mostlyai-engine"
-version = "1.1.3"
+version = "1.1.4"
 description = "Synthetic Data Engine"
 authors = [{ name = "MOSTLY AI", email = "dev@mostly.ai" }]
 requires-python = ">=3.10"
@@ -33,12 +33,13 @@ dependencies = [
     "psutil>=5.9.5,<6",  # upgrade when colab psutil is updated
     "tokenizers>=0.21.0,<0.22",
     "transformers>=4.45.2,<5",
-    "datasets>=2.20.0,<3",
-    "accelerate>=0.31.0,<0.32",
-    "peft>=0.11.1,<0.12",
+    "datasets>=3.0.0,<4",
+    "accelerate>=0.31.0",
+    "peft>=0.12.0",
+    "huggingface-hub>=0.28.0,<0.29",
     "opacus>=1.5.2,<2",
     "formatron>=0.4.11,<0.5",
-    "json-repair>=0.30.0,<0.31",
+    "json-repair>=0.30.0",
     "rich>=13.9.4,<14",
     "torch>=2.5.1; sys_platform == 'darwin'",
 ]