Skip to content

Commit

Permalink
removed hera deps..
Browse files Browse the repository at this point in the history
  • Loading branch information
borg committed Oct 16, 2024
1 parent ff63534 commit b265de9
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 22 deletions.
1 change: 0 additions & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,6 @@ dev-dependencies = [
"uv>=0.4.12",
"lorem",
"tox-uv>=1.13.0",
"hera"
]

[project.optional-dependencies]
Expand Down
21 changes: 0 additions & 21 deletions tests/integration/test_gpu_availability.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,5 @@
import pytest
import torch
from hera.workflows import DAG, Task, Workflow, script, WorkflowsService


SELECTOR_ARGO_SERVER_URL = "http://localhost:2746"
SELECTOR_SERVICE_ACCOUNT = "htrflow-service-account"
Expand All @@ -12,22 +10,3 @@ def test_gpu_availability():
assert torch.cuda.is_available(), "CUDA GPU is not available"
print(f"CUDA available: {torch.cuda.is_available()}")
print(f"Number of GPUs: {torch.cuda.device_count()}")


@script(image="python:3.12")
def echo(message):
print(message)


with Workflow(
generate_name="dag-diamond-",
service_account_name=SELECTOR_SERVICE_ACCOUNT,
workflows_service=WorkflowsService(host=SELECTOR_ARGO_SERVER_URL),
entrypoint="diamond",
) as w:
with DAG(name="diamond"):
A = Task(name="A", template=echo, arguments={"message": "A"})
B = Task(name="B", template=echo, arguments={"message": "B"})
A >> B

w.submit()

0 comments on commit b265de9

Please sign in to comment.