datatailr 0.1.2__tar.gz → 0.1.3__tar.gz

{datatailr-0.1.2 → datatailr-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datatailr
-Version: 0.1.2
+Version: 0.1.3
 Summary: The datatailr package
 Author-email: Datatailr <pypi@datatailr.com>
 License-Expression: MIT

{datatailr-0.1.2 → datatailr-0.1.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "datatailr"
-version = "0.1.2"
+version = "0.1.3"
 description = "The datatailr package"
 readme = "README.md"
 requires-python = ">=3.7"

{datatailr-0.1.2 → datatailr-0.1.3}/src/datatailr/scheduler/batch.py

@@ -14,7 +14,7 @@ import contextvars
 import json
 import os
 from functools import reduce
-from typing import List, Optional, Sequence, Set, Tuple, Union
+from typing import Dict, List, Optional, Sequence, Set, Tuple, Union
 
 from datatailr import Image
 from datatailr.errors import BatchJobError
@@ -31,7 +31,6 @@ from datatailr.scheduler.base import (
 from datatailr.scheduler.constants import DEFAULT_TASK_CPU, DEFAULT_TASK_MEMORY
 from datatailr.utils import is_dt_installed
 
-__BATCH_JOB_NAMES__ = {}
 __DAG_CONTEXT__: contextvars.ContextVar = contextvars.ContextVar("dag_context")
 logger = DatatailrLogger(os.path.abspath(__file__)).get_logger()
 
@@ -103,7 +102,7 @@ class BatchJob:
             raise MissingDagError()
         self.__id = dag.next_job_id
         self.dag = dag
-        __BATCH_JOB_NAMES__[self.name] = self.__id
+        self.dag.__BATCH_JOB_NAMES__[self.name] = self.__id
         self.dependencies = self.translate_dependencies()
         assert all(
             isinstance(dep, int) for dep in self.dependencies
@@ -116,10 +115,10 @@ class BatchJob:
 
         :param name: The alias name to set.
         """
-        if name in __BATCH_JOB_NAMES__:
+        if name in self.dag.__BATCH_JOB_NAMES__:
             raise DuplicateJobNameError(name)
-        assert __BATCH_JOB_NAMES__.pop(self.name) == self.__id
-        __BATCH_JOB_NAMES__[name] = self.__id
+        assert self.dag.__BATCH_JOB_NAMES__.pop(self.name) == self.__id
+        self.dag.__BATCH_JOB_NAMES__[name] = self.__id
         self.name = name
         return self
 
@@ -162,7 +161,10 @@ class BatchJob:
                 raise TypeError(f"Unsupported dependency type: {type(dep)}")
 
         return set(
-            [__BATCH_JOB_NAMES__[get_dependency_name(dep)] for dep in self.dependencies]
+            [
+                self.dag.__BATCH_JOB_NAMES__[get_dependency_name(dep)]
+                for dep in self.dependencies
+            ]
         )
 
     def __add_dependency__(self, other):
@@ -240,6 +242,7 @@ class Batch(Job):
         self.__jobs: List[BatchJob] = []
         self._auto_run = False
         self.__next_job_id = next_batch_job_id()
+        self.__BATCH_JOB_NAMES__: Dict[str, int] = {}
 
     @property
     def next_job_id(self):

{datatailr-0.1.2 → datatailr-0.1.3}/src/datatailr.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datatailr
-Version: 0.1.2
+Version: 0.1.3
 Summary: The datatailr package
 Author-email: Datatailr <pypi@datatailr.com>
 License-Expression: MIT