Skip to content

Commit

Permalink
[CLEANUP TORCH]
Browse files Browse the repository at this point in the history
  • Loading branch information
kyegomez committed Dec 14, 2024
1 parent cf43687 commit b73e4ee
Show file tree
Hide file tree
Showing 12 changed files with 1,160 additions and 215 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/code_quality_control.yml
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ jobs:
run: pip install --no-cache-dir -r requirements.txt

- name: Find Python files
run: find swarms_torch -name "*.py" -type f -exec autopep8 --in-place --aggressive --aggressive {} +
run: find clusterops -name "*.py" -type f -exec autopep8 --in-place --aggressive --aggressive {} +

- name: Push changes
uses: ad-m/github-push-action@master
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/lints.yml
Original file line number Diff line number Diff line change
Expand Up @@ -22,4 +22,4 @@ jobs:
run: pip install --no-cache-dir -r requirements.txt

- name: Run linters
run: pylint swarms_torch
run: pylint clusterops
2 changes: 1 addition & 1 deletion .github/workflows/pr_request_checks.yml
Original file line number Diff line number Diff line change
Expand Up @@ -24,4 +24,4 @@ jobs:
- name: Run tests and checks
run: |
pytest tests/
pylint swarms_torch
pylint clusterops
2 changes: 1 addition & 1 deletion .github/workflows/pull-request-links.yml
Original file line number Diff line number Diff line change
Expand Up @@ -15,4 +15,4 @@ jobs:
steps:
- uses: readthedocs/actions/preview@v1
with:
project-slug: swarms_torch
project-slug: clusterops
10 changes: 5 additions & 5 deletions clusterops/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,13 +5,15 @@
execute_on_gpu,
execute_on_cpu,
execute_on_multiple_gpus,
distributed_execute_on_gpus,
execute_with_all_cpu_cores

)
from clusterops.profiling_exec import (
monitor_resources,
profile_execution,
distributed_execute_on_gpus,
)
from clusterops.gpu_scheduler import GPUJobExecutor, JobScheduler, gpu_scheduler


__all__ = [
"list_available_cpus",
Expand All @@ -21,9 +23,7 @@
"execute_on_multiple_gpus",
"monitor_resources",
"profile_execution",
"GPUJobExecutor",
"execute_on_cpu",
"JobScheduler",
"gpu_scheduler",
"distributed_execute_on_gpus",
"execute_with_all_cpu_cores",
]
Loading

0 comments on commit b73e4ee

Please sign in to comment.