From 947b7eb9a0b2c2bd836095d6579236379001cc3e Mon Sep 17 00:00:00 2001 From: Yadu Babuji Date: Mon, 25 Nov 2024 16:58:47 -0600 Subject: [PATCH] Fixes to tests --- .github/workflows/gce_test.yaml | 8 +++----- globus_compute_executor/executor.py | 7 +++---- 2 files changed, 6 insertions(+), 9 deletions(-) diff --git a/.github/workflows/gce_test.yaml b/.github/workflows/gce_test.yaml index 697e1a6..6663aa6 100644 --- a/.github/workflows/gce_test.yaml +++ b/.github/workflows/gce_test.yaml @@ -61,17 +61,15 @@ jobs: - name: mypy run: | tox -e mypy - - name: make test + + - name: run tests env: GLOBUS_COMPUTE_CLIENT_ID: ${{ secrets.GLOBUS_COMPUTE_CLIENT_ID }} GLOBUS_COMPUTE_CLIENT_SECRET: ${{ secrets.GLOBUS_COMPUTE_SECRET_KEY }} run: | - source .venv/bin/activate export GLOBUS_COMPUTE_ENDPOINT=$(globus-compute-endpoint list | grep default | cut -c 3-38) echo "GLOBUS_COMPUTE_ENDPOINT = $GLOBUS_COMPUTE_ENDPOINT" - - tox -e mypy - + tox -e py - name: stop globus_compute_endpoint env: diff --git a/globus_compute_executor/executor.py b/globus_compute_executor/executor.py index 611e9bf..da57a3a 100644 --- a/globus_compute_executor/executor.py +++ b/globus_compute_executor/executor.py @@ -1,5 +1,6 @@ +from __future__ import annotations + import copy -import uuid from concurrent.futures import Future from typing import Any, Callable @@ -7,8 +8,6 @@ from parsl.executors.base import ParslExecutor from parsl.utils import RepresentationMixin -UUID_LIKE_T = uuid.UUID | str - class GlobusComputeExecutor(ParslExecutor, RepresentationMixin): """GlobusComputeExecutor is a plug-in for Parsl that enables remote task submission @@ -76,7 +75,7 @@ def submit( func: Callable, resource_specification: dict[str, Any], *args: Any, - **kwargs: Any + **kwargs: Any, ) -> Future: """Submit func to globus-compute