datatorch 0.6.1__tar.gz → 0.6.2__tar.gz

{datatorch-0.6.1 → datatorch-0.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datatorch
-Version: 0.6.1
+Version: 0.6.2
 Summary: A CLI and library for interacting with DataTorch.
 Home-page: https://github.com/datatorch/python
 Author: DataTorch

{datatorch-0.6.1 → datatorch-0.6.2}/datatorch/agent/client.py

@@ -78,9 +78,7 @@ class AgentApiClient(object):
             }
         """ % STEP_DISPATCH_FIELDS)
         # fmt: on
-        return cast(
-            AsyncGenerator[AgentStepRequest, None], self.session.subscribe(sub)
-        )
+        return cast(AsyncGenerator[AgentStepRequest, None], self.session.subscribe(sub))
 
     def agent_step_cancels(self):
         """Subscribe to stop signals for steps this agent is running.
@@ -156,9 +154,7 @@ class AgentApiClient(object):
             step_input["renderedInput"] = rendered_input
         if error_message is not None:
             step_input["errorMessage"] = error_message
-        result = await self.execute(
-            mutate, params={"id": step_id, "input": step_input}
-        )
+        result = await self.execute(mutate, params={"id": step_id, "input": step_input})
         return bool(result.get("completed"))
 
     def initial_metrics(self, metrics):

{datatorch-0.6.1 → datatorch-0.6.2}/datatorch/agent/pipelines/resolver.py

@@ -20,6 +20,7 @@ outputs are supplied and such a reference resolves as unresolvable, i.e.
 raises rather than passing a literal ``${{ }}`` to the action. A yaml that
 uses job dependencies is a server-executed workflow.
 """
+
 import json
 import re
 from typing import Any, Dict, List, Set, Tuple
@@ -29,7 +30,9 @@ from typing import Any, Dict, List, Set, Tuple
 # The name group is ``[^}]+?`` (not ``.+?``) so it can't backtrack across a
 # ``}}`` boundary — a string that both starts and ends with a ref must not be
 # mistaken for one spanning ref.
-STEP_REF = r"\$\{\{\s*steps\.(?P<stepName>[^}]+?)\.outputs?\.(?P<stepKey>[\w.\-]+)\s*\}\}"
+STEP_REF = (
+    r"\$\{\{\s*steps\.(?P<stepName>[^}]+?)\.outputs?\.(?P<stepKey>[\w.\-]+)\s*\}\}"
+)
 INPUT_REF = r"\$\{\{\s*inputs?\.(?P<inputKey>[\w.\-]+)\s*\}\}"
 JOB_REF = r"\$\{\{\s*jobs\.(?P<jobName>[^}]+?)\.outputs?\.(?P<jobKey>[\w.\-]+)\s*\}\}"
 
@@ -123,9 +126,7 @@ def _resolve(
         ]
     if isinstance(value, dict):
         return {
-            k: _resolve(
-                v, outputs_by_name, trigger_input, job_outputs_by_name, misses
-            )
+            k: _resolve(v, outputs_by_name, trigger_input, job_outputs_by_name, misses)
             for k, v in value.items()
         }
     return value

{datatorch-0.6.1 → datatorch-0.6.2}/datatorch/agent/pipelines/step/step.py

@@ -25,9 +25,7 @@ class Step(object):
     """
 
     @classmethod
-    def from_dispatch(
-        cls, dispatch: "AgentStepDispatch", api: "AgentApiClient" = None
-    ):
+    def from_dispatch(cls, dispatch: "AgentStepDispatch", api: "AgentApiClient" = None):
         return cls(
             id=dispatch.get("stepId"),
             action=dispatch.get("action", ""),

{datatorch-0.6.1 → datatorch-0.6.2}/datatorch/agent/pipelines/template.py

@@ -17,9 +17,7 @@ if typing.TYPE_CHECKING:
 # data (trigger payloads, fetched step outputs); splicing them into a shell
 # string is command injection, so shell/cmd runners must read them from
 # $INPUT_<NAME> env vars instead.
-_INPUT_REF_IN_TEMPLATE = re.compile(
-    r"\$\{[{%][^}]*?(?<![.\w])(?:input|variable)(?!\w)"
-)
+_INPUT_REF_IN_TEMPLATE = re.compile(r"\$\{[{%][^}]*?(?<![.\w])(?:input|variable)(?!\w)")
 
 
 class InputInjectionError(Exception):
@@ -146,9 +144,7 @@ class Variables(object):
                 "policy; see docs/Pipelines.md.)"
             )
         context = {
-            k: v
-            for k, v in self.variables.items()
-            if k not in ("input", "variable")
+            k: v for k, v in self.variables.items() if k not in ("input", "variable")
         }
         return self._template(string).render({**global_variables, **context})
 

{datatorch-0.6.1 → datatorch-0.6.2}/datatorch.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datatorch
-Version: 0.6.1
+Version: 0.6.2
 Summary: A CLI and library for interacting with DataTorch.
 Home-page: https://github.com/datatorch/python
 Author: DataTorch

{datatorch-0.6.1 → datatorch-0.6.2}/setup.py

@@ -33,7 +33,7 @@ requirements = [
 
 setup(
     name="datatorch",
-    version="0.6.1",
+    version="0.6.2",
     description="A CLI and library for interacting with DataTorch.",
     author="DataTorch",
     author_email="support@datatorch.io",