mostlyai-engine 1.4.1__tar.gz → 1.4.2__tar.gz

{mostlyai_engine-1.4.1 → mostlyai_engine-1.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mostlyai-engine
-Version: 1.4.1
+Version: 1.4.2
 Summary: Synthetic Data Engine
 Project-URL: homepage, https://github.com/mostly-ai/mostlyai-engine
 Project-URL: repository, https://github.com/mostly-ai/mostlyai-engine
@@ -35,17 +35,17 @@ Requires-Dist: pandas~=2.2.0
 Requires-Dist: peft>=0.12.0
 Requires-Dist: psutil<6,>=5.9.5
 Requires-Dist: pyarrow>=16.0.0
-Requires-Dist: setuptools<76,>=75.0.0
+Requires-Dist: setuptools>=77.0.3
 Requires-Dist: tokenizers>=0.21.0
 Requires-Dist: torch<2.6.1,>=2.6.0
 Requires-Dist: torchaudio<2.6.1,>=2.6.0
 Requires-Dist: torchvision<0.21.1,>=0.21.0
 Requires-Dist: transformers>=4.51.0
-Requires-Dist: xgrammar>=0.1.16
+Requires-Dist: xgrammar>=0.1.18
 Provides-Extra: gpu
-Requires-Dist: bitsandbytes<0.43,>=0.42.0; (sys_platform == 'darwin') and extra == 'gpu'
-Requires-Dist: bitsandbytes<0.46,>=0.43.3; (sys_platform == 'linux') and extra == 'gpu'
-Requires-Dist: vllm==0.8.3; (sys_platform == 'linux' or sys_platform == 'darwin') and extra == 'gpu'
+Requires-Dist: bitsandbytes==0.42.0; (sys_platform == 'darwin') and extra == 'gpu'
+Requires-Dist: bitsandbytes>=0.45.5; (sys_platform == 'linux') and extra == 'gpu'
+Requires-Dist: vllm==0.8.5.post1; (sys_platform == 'linux' or sys_platform == 'darwin') and extra == 'gpu'
 Description-Content-Type: text/markdown
 
 # Synthetic Data Engine 💎

{mostlyai_engine-1.4.1 → mostlyai_engine-1.4.2}/mostlyai/engine/__init__.py

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

{mostlyai_engine-1.4.1 → mostlyai_engine-1.4.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mostlyai-engine"
-version = "1.4.1"
+version = "1.4.2"
 description = "Synthetic Data Engine"
 authors = [{ name = "MOSTLY AI", email = "dev@mostly.ai" }]
 requires-python = ">=3.10"
@@ -25,7 +25,7 @@ classifiers = [
 ]
 
 dependencies = [
-    "setuptools>=75.0.0,<76",
+    "setuptools>=77.0.3",  # similar to vllm 0.8.5.post1
     "numpy>=1.26.3",
     "pandas~=2.2.0",
     "pyarrow>=16.0.0",
@@ -38,18 +38,18 @@ dependencies = [
     "peft>=0.12.0",
     "huggingface-hub[hf-xet]>=0.30.2",
     "opacus>=1.5.2",  # switch to 1.5.4 (once released) to allow numpy 2
-    "xgrammar>=0.1.16",  # for vllm 0.8.3 compatibility
+    "xgrammar>=0.1.18",  # for vllm 0.8.5.post1 compatibility
     "json-repair>=0.30.0",
     "torch>=2.6.0,<2.6.1",
-    "torchaudio>=2.6.0,<2.6.1",  # for vllm 0.8.3 compatibility
-    "torchvision>=0.21.0,<0.21.1"  # for vllm 0.8.3 compatibility
+    "torchaudio>=2.6.0,<2.6.1",  # for vllm 0.8.5.post1 compatibility
+    "torchvision>=0.21.0,<0.21.1"  # for vllm 0.8.5.post1 compatibility
 ]
 
 [project.optional-dependencies]
 gpu = [
-    "bitsandbytes>=0.42.0,<0.43; sys_platform == 'darwin'",
-    "bitsandbytes>=0.43.3,<0.46; sys_platform == 'linux'",
-    "vllm==0.8.3; sys_platform == 'linux' or sys_platform == 'darwin'",
+    "bitsandbytes==0.42.0; sys_platform == 'darwin'",
+    "bitsandbytes>=0.45.5; sys_platform == 'linux'",
+    "vllm==0.8.5.post1; sys_platform == 'linux' or sys_platform == 'darwin'",
 ]
 
 [dependency-groups]