Skip to content

Commit

Permalink
Fix: resolved dependencies
Browse files Browse the repository at this point in the history
  • Loading branch information
jbarnes850 committed Nov 19, 2024
1 parent a0b9940 commit 0ae8dec
Show file tree
Hide file tree
Showing 2 changed files with 81 additions and 81 deletions.
84 changes: 42 additions & 42 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -23,54 +23,54 @@ classifiers = [
requires-python = ">=3.11,<3.12"

dependencies = [
"mlx==0.20.0",
"numpy==1.24.3",
"transformers==4.36.2",
"safetensors==0.4.1",
"datasets==2.15.0",
"torch==2.1.2",
"mpi4py==3.1.5",
"paramiko==3.3.1",
"pyzmq==25.1.1",
"asyncssh==2.14.1",
"iperf3==0.1.11",
"aiofiles==23.2.1",
"aiosqlite==0.19.0",
"xxhash==3.4.1",
"fastapi==0.104.1",
"uvicorn==0.24.0",
"pydantic==2.5.2",
"psutil==5.9.6",
"plotly==5.18.0",
"dash==2.14.1",
"dash-core-components==2.0.0",
"dash-html-components==2.0.0",
"pandas==2.1.3",
"tqdm==4.66.1",
"pyyaml==6.0.1",
"metal-sdk==2.5.1",
"huggingface-hub==0.19.4",
"mlx>=0.20.0",
"numpy>=1.24.0",
"transformers>=4.36.0",
"safetensors>=0.4.0",
"datasets>=2.15.0",
"torch>=2.1.0",
"mpi4py>=3.1.0",
"paramiko>=3.3.0",
"pyzmq>=25.1.0",
"asyncssh>=2.14.0",
"iperf3>=0.1.11",
"aiofiles>=23.2.0",
"aiosqlite>=0.19.0",
"xxhash>=3.4.0",
"fastapi>=0.104.0",
"uvicorn>=0.24.0",
"pydantic>=2.5.0",
"psutil>=5.9.0",
"plotly>=5.18.0",
"dash>=2.14.0",
"dash-core-components>=2.0.0",
"dash-html-components>=2.0.0",
"pandas>=2.1.0",
"tqdm>=4.66.0",
"pyyaml>=6.0.0",
"metal-sdk>=2.5.0",
"huggingface-hub>=0.19.0",
]

[project.optional-dependencies]
dev = [
"pytest==7.4.3",
"pytest-asyncio==0.21.1",
"pytest-cov==4.1.0",
"black==23.11.0",
"isort==5.12.0",
"mypy==1.7.1",
"pylint==3.0.2",
"types-PyYAML==6.0.12.12",
"types-tqdm==4.65.0.1",
"types-paramiko==3.3.1",
"types-psutil==5.9.0",
"types-aiofiles==23.2.0",
"pytest>=7.4.0",
"pytest-asyncio>=0.21.0",
"pytest-cov>=4.1.0",
"black>=23.11.0",
"isort>=5.12.0",
"mypy>=1.7.0",
"pylint>=3.0.0",
"types-PyYAML>=6.0.0",
"types-tqdm>=4.65.0",
"types-paramiko>=3.5.0.20240918",
"types-psutil>=5.9.0",
"types-aiofiles>=23.2.0",
]
docs = [
"sphinx==7.2.6",
"sphinx-rtd-theme==1.3.0",
"sphinx-autodoc-typehints==1.25.2",
"sphinx>=7.2.0",
"sphinx-rtd-theme>=1.3.0",
"sphinx-autodoc-typehints>=1.25.0",
]

[tool.setuptools.packages.find]
Expand Down
78 changes: 39 additions & 39 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -1,54 +1,54 @@
# Core ML Dependencies
mlx==0.20.0
numpy==1.24.3
transformers==4.36.2
safetensors==0.4.1
datasets==2.15.0
torch==2.1.2
mlx>=0.20.0
numpy>=1.24.0
transformers>=4.36.0
safetensors>=0.4.0
datasets>=2.15.0
torch>=2.1.0

# Distributed Training
mpi4py==3.1.5
paramiko==3.3.1
pyzmq==25.1.1
asyncssh==2.14.1 # Used in network_utils
iperf3==0.1.11 # For network performance testing
mpi4py>=3.1.0
paramiko>=3.3.0
pyzmq>=25.1.0
asyncssh>=2.14.0 # Used in network_utils
iperf3>=0.1.11 # For network performance testing

# Storage & IO
aiofiles==23.2.1 # Used in checkpoint_utils
aiosqlite==0.19.0 # Used in checkpoint_utils
xxhash==3.4.1 # Used in data_utils
aiofiles>=23.2.0 # Used in checkpoint_utils
aiosqlite>=0.19.0 # Used in checkpoint_utils
xxhash>=3.4.0 # Used in data_utils

# API & Serving
fastapi==0.104.1
uvicorn==0.24.0
pydantic==2.5.2
fastapi>=0.104.0
uvicorn>=0.24.0
pydantic>=2.5.0

# Utilities
tqdm==4.66.1
pyyaml==6.0.1
psutil==5.9.6
huggingface-hub==0.19.4
metal-sdk==2.5.1
tqdm>=4.66.0
pyyaml>=6.0.0
psutil>=5.9.0
huggingface-hub>=0.19.0
metal-sdk>=2.5.0

# Monitoring & Visualization
plotly==5.18.0
dash==2.14.1
dash-core-components==2.0.0
dash-html-components==2.0.0
pandas==2.1.3
plotly>=5.18.0
dash>=2.14.0
dash-core-components>=2.0.0
dash-html-components>=2.0.0
pandas>=2.1.0

# Testing & Development
pytest==7.4.3
pytest-asyncio==0.21.1
pytest-cov==4.1.0
black==23.11.0
isort==5.12.0
mypy==1.7.1
pylint==3.0.2
pytest>=7.4.0
pytest-asyncio>=0.21.0
pytest-cov>=4.1.0
black>=23.11.0
isort>=5.12.0
mypy>=1.7.0
pylint>=3.0.0

# Type Stubs
types-PyYAML==6.0.12.12
types-tqdm==4.65.0.1
types-paramiko==3.3.1
types-psutil==5.9.0
types-aiofiles==23.2.0
types-PyYAML>=6.0.0
types-tqdm>=4.65.0
types-paramiko>=3.5.0.20240918 # Updated to latest available version
types-psutil>=5.9.0
types-aiofiles>=23.2.0

0 comments on commit 0ae8dec

Please sign in to comment.