mostlyai-engine 1.2.4__tar.gz → 1.3.1__tar.gz

{mostlyai_engine-1.2.4 → mostlyai_engine-1.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mostlyai-engine
-Version: 1.2.4
+Version: 1.3.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
@@ -37,25 +37,15 @@ Requires-Dist: psutil<6,>=5.9.5
 Requires-Dist: pyarrow>=16.0.0
 Requires-Dist: setuptools<76,>=75.0.0
 Requires-Dist: tokenizers>=0.21.0
-Requires-Dist: torch==2.6.0; sys_platform == 'linux'
-Requires-Dist: torch>=2.6.0; sys_platform == 'darwin'
-Requires-Dist: torchaudio==2.6.0; sys_platform == 'linux'
-Requires-Dist: torchvision==0.21.0; sys_platform == 'linux'
+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
-Provides-Extra: cpu
-Requires-Dist: torch==2.6.0+cpu; (sys_platform == 'linux') and extra == 'cpu'
-Requires-Dist: torch==2.6.0; (sys_platform != 'linux') and extra == 'cpu'
-Requires-Dist: torchaudio==2.6.0+cpu; (sys_platform == 'linux') and extra == 'cpu'
-Requires-Dist: torchvision==0.21.0+cpu; (sys_platform == 'linux') and extra == 'cpu'
 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: torch==2.6.0; extra == 'gpu'
-Requires-Dist: torchaudio==2.6.0; extra == 'gpu'
-Requires-Dist: torchvision==0.21.0; extra == 'gpu'
-Requires-Dist: vllm<=0.9,>=0.7.3; (sys_platform == 'darwin') and extra == 'gpu'
-Requires-Dist: vllm==0.8.3; (sys_platform == 'linux') and extra == 'gpu'
+Requires-Dist: vllm==0.8.3; (sys_platform == 'linux' or sys_platform == 'darwin') and extra == 'gpu'
 Description-Content-Type: text/markdown
 
 # Synthetic Data Engine 💎
@@ -93,17 +83,17 @@ The latest release of `mostlyai-engine` can be installed via pip:
 pip install -U mostlyai-engine
 ```
 
-or alternatively for a CPU-only setup:
+or alternatively for a GPU setup (needed for LLM finetuning and inference):
 ```bash
-pip install -U 'mostlyai-engine[cpu]' --extra-index-url https://download.pytorch.org/whl/cpu
+pip install -U 'mostlyai-engine[gpu]'
 ```
 
-or alternatively for a GPU setup (needed for LLM finetuning):
+On Linux, one can explicitly install the CPU-only variant of torch together with `mostlyai-engine`:
+
 ```bash
-pip install -U 'mostlyai-engine[gpu]'
+pip install -U torch==2.6.0+cpu torchvision==0.21.0+cpu mostlyai-engine --extra-index-url https://download.pytorch.org/whl/cpu
 ```
 
-
 ## Quick start
 
 ### Tabular Model: flat data, without context

{mostlyai_engine-1.2.4 → mostlyai_engine-1.3.1}/README.md

@@ -33,17 +33,17 @@ The latest release of `mostlyai-engine` can be installed via pip:
 pip install -U mostlyai-engine
 ```
 
-or alternatively for a CPU-only setup:
+or alternatively for a GPU setup (needed for LLM finetuning and inference):
 ```bash
-pip install -U 'mostlyai-engine[cpu]' --extra-index-url https://download.pytorch.org/whl/cpu
+pip install -U 'mostlyai-engine[gpu]'
 ```
 
-or alternatively for a GPU setup (needed for LLM finetuning):
+On Linux, one can explicitly install the CPU-only variant of torch together with `mostlyai-engine`:
+
 ```bash
-pip install -U 'mostlyai-engine[gpu]'
+pip install -U torch==2.6.0+cpu torchvision==0.21.0+cpu mostlyai-engine --extra-index-url https://download.pytorch.org/whl/cpu
 ```
 
-
 ## Quick start
 
 ### Tabular Model: flat data, without context

{mostlyai_engine-1.2.4 → mostlyai_engine-1.3.1}/mostlyai/engine/__init__.py

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

{mostlyai_engine-1.2.4 → mostlyai_engine-1.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mostlyai-engine"
-version = "1.2.4"
+version = "1.3.1"
 description = "Synthetic Data Engine"
 authors = [{ name = "MOSTLY AI", email = "dev@mostly.ai" }]
 requires-python = ">=3.10"
@@ -40,27 +40,16 @@ dependencies = [
     "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
     "json-repair>=0.30.0",
-    "torch>=2.6.0; sys_platform == 'darwin'",
-    "torch==2.6.0; sys_platform == 'linux'",  # for vllm 0.8.3 compatibility
-    "torchaudio==2.6.0; sys_platform == 'linux'",  # for vllm 0.8.3 compatibility
-    "torchvision==0.21.0; sys_platform == 'linux'"  # for vllm 0.8.3 compatibility
+    "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
 ]
 
 [project.optional-dependencies]
 gpu = [
-    "torch==2.6.0",  # for vllm 0.8.3 compatibility
-    "torchaudio==2.6.0",  # for vllm 0.8.3 compatibility
-    "torchvision==0.21.0",  # for vllm 0.8.3 compatibility
     "bitsandbytes>=0.42.0,<0.43; sys_platform == 'darwin'",
-    "vllm>=0.7.3,<=0.9; sys_platform == 'darwin'",
     "bitsandbytes>=0.43.3,<0.46; sys_platform == 'linux'",
-    "vllm==0.8.3; sys_platform == 'linux'",
-]
-cpu = [
-    "torch==2.6.0+cpu; sys_platform == 'linux'",  # for vllm 0.8.3 compatibility
-    "torchaudio==2.6.0+cpu; sys_platform == 'linux'",  # for vllm 0.8.3 compatibility
-    "torchvision==0.21.0+cpu; sys_platform == 'linux'",  # for vllm 0.8.3 compatibility
-    "torch==2.6.0; sys_platform != 'linux'",
+    "vllm==0.8.3; sys_platform == 'linux' or sys_platform == 'darwin'",
 ]
 
 [dependency-groups]
@@ -88,28 +77,6 @@ documentation = "https://mostly-ai.github.io/mostlyai-engine/"
 
 [tool.uv]
 default-groups = ["dev", "docs"]
-conflicts = [
-  [
-    { extra = "cpu" },
-    { extra = "gpu" },
-  ],
-]
-
-[[tool.uv.index]]
-name = "pytorch-cpu"
-url = "https://download.pytorch.org/whl/cpu"
-explicit = true
-
-[tool.uv.sources]
-torch = [
-  { index = "pytorch-cpu", extra = "cpu", marker = "sys_platform == 'linux'"},
-]
-torchaudio = [
-  { index = "pytorch-cpu", extra = "cpu", marker = "sys_platform == 'linux'"},
-]
-torchvision = [
-  { index = "pytorch-cpu", extra = "cpu", marker = "sys_platform == 'linux'"},
-]
 
 [tool.hatch.build.targets.sdist]
 include = ["mostlyai/engine"]