Add maximum_active_task_schedulers method to Env

PiperOrigin-RevId: 660154575
tensorflow · Aug 7, 2024 · cd5bddd · cd5bddd
1 parent 5e90c67
commit cd5bddd
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tfx/orchestration/experimental/core/env.py b/tfx/orchestration/experimental/core/env.py
@@ -158,6 +158,10 @@ def get_status_code_from_exception(
       Returns None if the exception is not a known type.
     """
 
+  @abc.abstractmethod
+  def maximum_concurrent_task_schedulers(self) -> int:
+    """Returns the maximum number of concurrent task schedulers."""
+
 
 class _DefaultEnv(Env):
   """Default environment."""
@@ -244,6 +248,9 @@ def get_status_code_from_exception(
   ) -> Optional[int]:
     return None
 
+  def maximum_concurrent_task_schedulers(self) -> int:
+    return 1
+
 
 _ENV = _DefaultEnv()