jobflow 0.1.9__tar.gz → 0.1.11__tar.gz

{jobflow-0.1.9 → jobflow-0.1.11}/PKG-INFO

@@ -1,12 +1,14 @@
 Metadata-Version: 2.1
 Name: jobflow
-Version: 0.1.9
-Summary: jobflow is library for writing computational workflows
-Home-page: https://materialsproject.github.io/jobflow
-Author: Alex Ganose
-Author-email: alexganose@googlemail.com
+Version: 0.1.11
+Summary: jobflow is a library for writing computational workflows
+Author-email: Alex Ganose <alexganose@gmail.com>
 License: modified BSD
-Platform: UNKNOWN
+Project-URL: homepage, https://materialsproject.github.io/jobflow/
+Project-URL: repository, https://github.com/materialsproject/jobflow
+Project-URL: documentation, https://materialsproject.github.io/jobflow/
+Project-URL: changelog, https://github.com/materialsproject/jobflow/blob/main/CHANGELOG.md
+Keywords: high-throughput,workflow
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
@@ -22,19 +24,22 @@ Classifier: Topic :: Scientific/Engineering
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 Provides-Extra: docs
-Provides-Extra: tests
 Provides-Extra: dev
+Provides-Extra: tests
 Provides-Extra: vis
 Provides-Extra: fireworks
+Provides-Extra: strict
 License-File: LICENSE
 
 # jobflow
 
-<a href="https://github.com/materialsproject/jobflow/actions?query=workflow%3Atesting"><img alt="code coverage" src="https://img.shields.io/github/workflow/status/materialsproject/jobflow/testing?label=tests"></a>
+<a href="https://github.com/materialsproject/jobflow/actions?query=workflow%3Atesting"><img alt="code coverage" src="https://img.shields.io/github/actions/workflow/status/materialsproject/jobflow/testing.yml?branch=main&label=tests"></a>
 <a href="https://codecov.io/gh/materialsproject/jobflow/"><img alt="code coverage" src="https://img.shields.io/codecov/c/gh/materialsproject/jobflow/main"></a>
 <a href="https://pypi.org/project/jobflow"><img alt="pypi version" src="https://img.shields.io/pypi/v/jobflow?color=blue"></a>
 <img alt="supported python versions" src="https://img.shields.io/pypi/pyversions/jobflow">
 
+[Documentation](https://materialsproject.github.io/jobflow/) | [PyPI](https://pypi.org/project/jobflow/) | [GitHub](https://github.com/materialsproject/jobflow)
+
 Jobflow is a free, open-source library for writing and executing workflows. Complex
 workflows can be defined using simple python functions and executed locally or on
 arbitrary computing resources using the [FireWorks][fireworks] workflow manager.
@@ -104,7 +109,7 @@ the jobs is determined automatically and can be visualised using the flow graph.
 
 ## Installation
 
-The jobflow is a Python 3.7+ library and can be installed using pip.
+The jobflow is a Python 3.8+ library and can be installed using pip.
 
 ```bash
 pip install jobflow
@@ -157,5 +162,3 @@ Jobflow was designed by Alex Ganose, Anubhav Jain, Gian-Marco Rignanese, David W
 [dynamic-flows]: https://materialsproject.github.io/jobflow/tutorials/5-dynamic-flows.html
 [jobflow-database]: https://materialsproject.github.io/jobflow/tutorials/6-jobflow-database.html
 [jobflow-fireworks]: https://materialsproject.github.io/jobflow/tutorials/7-fireworks.html
-
-

{jobflow-0.1.9 → jobflow-0.1.11}/README.md

@@ -1,10 +1,12 @@
 # jobflow
 
-<a href="https://github.com/materialsproject/jobflow/actions?query=workflow%3Atesting"><img alt="code coverage" src="https://img.shields.io/github/workflow/status/materialsproject/jobflow/testing?label=tests"></a>
+<a href="https://github.com/materialsproject/jobflow/actions?query=workflow%3Atesting"><img alt="code coverage" src="https://img.shields.io/github/actions/workflow/status/materialsproject/jobflow/testing.yml?branch=main&label=tests"></a>
 <a href="https://codecov.io/gh/materialsproject/jobflow/"><img alt="code coverage" src="https://img.shields.io/codecov/c/gh/materialsproject/jobflow/main"></a>
 <a href="https://pypi.org/project/jobflow"><img alt="pypi version" src="https://img.shields.io/pypi/v/jobflow?color=blue"></a>
 <img alt="supported python versions" src="https://img.shields.io/pypi/pyversions/jobflow">
 
+[Documentation](https://materialsproject.github.io/jobflow/) | [PyPI](https://pypi.org/project/jobflow/) | [GitHub](https://github.com/materialsproject/jobflow)
+
 Jobflow is a free, open-source library for writing and executing workflows. Complex
 workflows can be defined using simple python functions and executed locally or on
 arbitrary computing resources using the [FireWorks][fireworks] workflow manager.
@@ -74,7 +76,7 @@ the jobs is determined automatically and can be visualised using the flow graph.
 
 ## Installation
 
-The jobflow is a Python 3.7+ library and can be installed using pip.
+The jobflow is a Python 3.8+ library and can be installed using pip.
 
 ```bash
 pip install jobflow

jobflow-0.1.11/pyproject.toml

@@ -0,0 +1,140 @@
+[build-system]
+requires = ["setuptools >= 42", "versioningit ~= 1.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "jobflow"
+description = "jobflow is a library for writing computational workflows"
+readme = "README.md"
+keywords = ["high-throughput", "workflow"]
+license = { text = "modified BSD" }
+authors = [{ name = "Alex Ganose", email = "alexganose@gmail.com" }]
+dynamic = ["version"]
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.8",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Development Status :: 2 - Pre-Alpha",
+    "Intended Audience :: Science/Research",
+    "Intended Audience :: System Administrators",
+    "Intended Audience :: Information Technology",
+    "Operating System :: OS Independent",
+    "Topic :: Other/Nonlisted Topic",
+    "Topic :: Database :: Front-Ends",
+    "Topic :: Scientific/Engineering",
+]
+requires-python = ">=3.8"
+dependencies = [
+    "monty>=2021.5.9",
+    "pydash",
+    "networkx",
+    "maggma>=0.38.1",
+    "pydantic",
+    "PyYAML",
+]
+
+[project.optional-dependencies]
+docs = [
+    "sphinx==6.1.3",
+    "furo==2022.12.7",
+    "myst_parser==1.0.0",
+    "ipython==8.11.0",
+    "nbsphinx==0.9.0",
+    "autodoc_pydantic==1.8.0",
+]
+dev = ["pre-commit>=2.12.1"]
+tests = ["pytest==7.2.2", "pytest-cov==4.0.0"]
+vis = ["matplotlib", "pydot"]
+fireworks = ["FireWorks"]
+strict = [
+    "monty==2022.9.9",
+    "networkx==3.0",
+    "pydash==6.0.2",
+    "maggma==0.50.3",
+    "pydantic==1.10.6",
+    "PyYAML==6.0",
+    "FireWorks==2.0.3",
+    "matplotlib==3.7.1",
+    "pydot==1.4.2",
+    "moto==4.1.4",
+]
+
+[project.urls]
+homepage = "https://materialsproject.github.io/jobflow/"
+repository = "https://github.com/materialsproject/jobflow"
+documentation = "https://materialsproject.github.io/jobflow/"
+changelog = "https://github.com/materialsproject/jobflow/blob/main/CHANGELOG.md"
+
+[tool.setuptools.package-data]
+jobflow = ["py.typed"]
+
+[tool.versioningit.vcs]
+method = "git"
+default-tag = "0.0.1"
+
+[tool.flake8]
+max-line-length = 88
+max-doc-length = 88
+select = "C, E, F, W, B"
+extend-ignore = "E203, W503, E501, F401, RST21"
+min-python-version = "3.8.0"
+docstring-convention = "numpy"
+rst-roles = "class, func, ref, obj"
+
+[tool.mypy]
+ignore_missing_imports = true
+no_strict_optional = true
+
+[tool.pytest.ini_options]
+filterwarnings = [
+    "ignore:.*POTCAR.*:UserWarning",
+    "ignore:.*magmom.*:UserWarning",
+    "ignore:.*is not gzipped.*:UserWarning",
+    "ignore:.*input structure.*:UserWarning",
+    "ignore::DeprecationWarning",
+]
+
+[tool.coverage.run]
+include = ["src/*"]
+parallel = true
+branch = true
+
+[tool.coverage.paths]
+source = ["src/"]
+
+[tool.coverage.report]
+skip_covered = true
+show_missing = true
+exclude_lines = [
+    '^\s*assert False(,|$)',
+    'if typing.TYPE_CHECKING:',
+    '^\s*@overload( |$)',
+]
+
+[tool.ruff]
+target-version = "py38"
+ignore-init-module-imports = true
+select = [
+    "B",   # flake8-bugbear
+    "C4",  # flake8-comprehensions
+    "D",   # pydocstyle
+    "E",   # pycodestyle
+    "F",   # pyflakes
+    "I",   # isort
+    "PLE", # pylint error
+    "PLW", # pylint warning
+    "Q",   # flake8-quotes
+    "RUF", # Ruff-specific rules
+    "SIM", # flake8-simplify
+    "TID", # tidy imports
+    "UP",  # pyupgrade
+    "W",   # pycodestyle
+    "YTT", # flake8-2020
+]
+pydocstyle.convention = "numpy"
+isort.known-first-party = ["jobflow"]
+
+[tool.ruff.per-file-ignores]
+"__init__.py" = ["F401"]
+"**/tests/*" = ["D"]

jobflow-0.1.11/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

{jobflow-0.1.9 → jobflow-0.1.11}/src/jobflow/core/flow.py

@@ -140,7 +140,7 @@ class Flow(MSONable):
         self.uuid = uuid
         self.hosts = hosts or []
 
-        self._jobs: tuple[Flow | Job, ...] = tuple()
+        self._jobs: tuple[Flow | Job, ...] = ()
         self.add_jobs(jobs)
         self.output = output
 
@@ -173,7 +173,8 @@ class Flow(MSONable):
         """
         Set the output of the Flow.
 
-        The output should be compatible with the list of Jobs/Flows contained in the Flow.
+        The output should be compatible with the list of Jobs/Flows contained in the
+        Flow.
 
         Parameters
         ----------
@@ -406,8 +407,9 @@ class Flow(MSONable):
         update
             The updates to apply.
         name_filter
-            A filter for the Maker name. Only Makers with a matching name will be updated.
-            Includes partial matches, e.g. "ad" will match a Maker with the name "adder".
+            A filter for the Maker name. Only Makers with a matching name will be
+            updated. Includes partial matches, e.g. "ad" will match a Maker with the
+            name "adder".
         class_filter
             A filter for the maker class. Only Makers with a matching class will be
             updated. Note the class filter will match any subclasses.
@@ -533,7 +535,8 @@ class Flow(MSONable):
             Use the dict mod language to apply updates. See :obj:`.DictMods` for more
             details.
         dynamic
-            The updates will be propagated to Jobs/Flows dynamically generated at runtime.
+            The updates will be propagated to Jobs/Flows dynamically generated at
+            runtime.
 
         Examples
         --------
@@ -587,7 +590,8 @@ class Flow(MSONable):
             Which attributes of the job config to set. Can be specified as one or more
             attributes specified by their name.
         dynamic
-            The updates will be propagated to Jobs/Flows dynamically generated at runtime.
+            The updates will be propagated to Jobs/Flows dynamically generated at
+            runtime.
 
         Examples
         --------
@@ -678,7 +682,7 @@ class Flow(MSONable):
             jobs = [jobs]
 
         job_ids = set(self.all_uuids)
-        hosts = [self.uuid] + self.hosts
+        hosts = [self.uuid, *self.hosts]
         for job in jobs:
             if job.host is not None and job.host != self.uuid:
                 raise ValueError(

{jobflow-0.1.9 → jobflow-0.1.11}/src/jobflow/core/job.py

@@ -49,8 +49,8 @@ class JobConfig(MSONable):
     response_manager_config
         The custom configuration to pass to a detour, addition, or replacement job.
         Using this kwarg will automatically take precedence over the behavior of
-        ``pass_manager_config`` such that a different configuration than ``manger_config``
-        can be passed to downstream jobs.
+        ``pass_manager_config`` such that a different configuration than
+        ``manger_config`` can be passed to downstream jobs.
 
     Returns
     -------
@@ -116,9 +116,9 @@ def job(method: Callable | None = None, **job_kwargs):
     OutputReference('abeb6f48-9b34-4698-ab69-e4dc2127ebe9')
 
     .. Note::
-        Because the task has not yet been run, the output value is :obj:`OutputReference`
-        object. References are automatically converted to their computed values
-        (resolved) when the task runs.
+        Because the task has not yet been run, the output value is an
+        :obj:`OutputReference` object. References are automatically converted to their
+        computed values (resolved) when the task runs.
 
     If a dictionary of values is returned, the values can be indexed in the usual
     way.
@@ -251,10 +251,11 @@ class Job(MSONable):
         The config setting for the job.
     hosts
         The list of UUIDs of the hosts containing the job. The object identified by one
-        UUID of the list should be contained in objects identified by its subsequent elements.
+        UUID of the list should be contained in objects identified by its subsequent
+        elements.
     metadata_updates
-        A list of updates for the metadata that will be applied to any Flow/Job generated
-        by the job.
+        A list of updates for the metadata that will be applied to any Flow/Job
+        generated by the job.
     config_updates
         A list of updates for the config that will be applied to any Flow/Job generated
         by the job.
@@ -323,7 +324,7 @@ class Job(MSONable):
 
         from jobflow.utils.find import contains_flow_or_job
 
-        function_args = tuple() if function_args is None else function_args
+        function_args = () if function_args is None else function_args
         function_kwargs = {} if function_kwargs is None else function_kwargs
         uuid = suuid() if uuid is None else uuid
         metadata = {} if metadata is None else metadata
@@ -584,12 +585,14 @@ class Job(MSONable):
                 pass_manager_config(response.replace, passed_config)
 
         try:
-            output = jsanitize(response.output, strict=True, enum_values=True)
-        except AttributeError:
+            output = jsanitize(
+                response.output, strict=True, enum_values=True, allow_bson=True
+            )
+        except AttributeError as err:
             raise RuntimeError(
                 "Job output contained an object that is not MSONable and therefore "
                 "could not be serialized."
-            )
+            ) from err
 
         save = {k: "output" if v is True else v for k, v in self._kwargs.items()}
         data = {
@@ -730,8 +733,9 @@ class Job(MSONable):
         update
             The updates to apply.
         name_filter
-            A filter for the Maker name. Only Makers with a matching name will be updated.
-            Includes partial matches, e.g. "ad" will match a Maker with the name "adder".
+            A filter for the Maker name. Only Makers with a matching name will be
+            updated. Includes partial matches, e.g. "ad" will match a Maker with the
+            name "adder".
         class_filter
             A filter for the maker class. Only Makers with a matching class will be
             updated. Note the class filter will match any subclasses.
@@ -802,7 +806,7 @@ class Job(MSONable):
                 nested=nested,
                 dict_mod=dict_mod,
             )
-            setattr(self, "function", getattr(maker, self.function.__name__))
+            self.function = getattr(maker, self.function.__name__)
         elif nested:
             # also look for makers in job args and kwargs
             new_args = []
@@ -874,12 +878,13 @@ class Job(MSONable):
             Use the dict mod language to apply updates. See :obj:`.DictMods` for more
             details.
         dynamic
-            The updates will be propagated to Jobs/Flows dynamically generated at runtime.
+            The updates will be propagated to Jobs/Flows dynamically generated at
+            runtime.
 
         Examples
         --------
-        Consider a simple job that makes use of a :obj:`Maker` to generate additional jobs
-        at runtime (see :obj:`Response` options for more details):
+        Consider a simple job that makes use of a :obj:`Maker` to generate additional
+        jobs at runtime (see :obj:`Response` options for more details):
 
         >>> @job
         ... def use_maker(maker):
@@ -902,12 +907,12 @@ class Job(MSONable):
         from jobflow.utils.dict_mods import apply_mod
 
         if dynamic:
-            dict_input = dict(
-                update=update,
-                name_filter=name_filter,
-                function_filter=function_filter,
-                dict_mod=dict_mod,
-            )
+            dict_input = {
+                "update": update,
+                "name_filter": name_filter,
+                "function_filter": function_filter,
+                "dict_mod": dict_mod,
+            }
             self.metadata_updates.append(dict_input)
 
         # unwrap the functions in case the job is a decorated one
@@ -957,7 +962,8 @@ class Job(MSONable):
             Which attributes of the job config to set. Can be specified as one or more
             attributes specified by their name.
         dynamic
-            The updates will be propagated to Jobs/Flows dynamically generated at runtime.
+            The updates will be propagated to Jobs/Flows dynamically generated at
+            runtime.
 
         Examples
         --------
@@ -976,9 +982,9 @@ class Job(MSONable):
         ... )
         >>> add_job.update_config(new_config)
 
-        To only update specific attributes, the ``attributes`` argument can be specified.
-        For example, the following will only update the "manager_config" attribute of the
-        job config.
+        To only update specific attributes, the ``attributes`` argument can be
+        specified. For example, the following will only update the "manager_config"
+        attribute of the job config.
 
         >>> add_job.update_config(new_config, attributes="manager_config")
 
@@ -1010,12 +1016,12 @@ class Job(MSONable):
         only be set for the `test_job` and not for the generated Jobs.
         """
         if dynamic:
-            dict_input = dict(
-                config=config,
-                name_filter=name_filter,
-                function_filter=function_filter,
-                attributes=attributes,
-            )
+            dict_input = {
+                "config": config,
+                "name_filter": name_filter,
+                "function_filter": function_filter,
+                "attributes": attributes,
+            }
             self.config_updates.append(dict_input)
 
         # unwrap the functions in case the job is a decorated one
@@ -1033,15 +1039,15 @@ class Job(MSONable):
 
         # if we get to here then we pass all the filters
         if isinstance(config, dict):
-            # convert dict specification to a JobConfig but set the attributes accordingly
+            # convert dict specification to a JobConfig but set the attributes
             if attributes is None:
                 attributes = list(config.keys())
 
             attributes = [attributes] if isinstance(attributes, str) else attributes
             if not set(attributes).issubset(set(config.keys())):
                 raise ValueError(
-                    "Specified attributes include a key that is not present in the config"
-                    " dictionary."
+                    "Specified attributes include a key that is not present in the "
+                    "config dictionary."
                 )
             config = JobConfig(**config)
 
@@ -1098,8 +1104,15 @@ class Job(MSONable):
             self.hosts.extend(hosts_uuids)
 
 
+# For type checking, the Response output type can be specified
+# in a type hint via this type variable.
+# For example, a signature `-> Response[int]` would require
+# that the Response.output is an int.
+T = typing.TypeVar("T")
+
+
 @dataclass
-class Response:
+class Response(typing.Generic[T]):
     """
     The :obj:`Response` contains the output, detours, and stop commands of a job.
 
@@ -1121,7 +1134,7 @@ class Response:
         Stop executing all remaining jobs.
     """
 
-    output: Any | None = None
+    output: T | None = None
     detour: jobflow.Flow | Job | list[Job] | list[jobflow.Flow] | None = None
     addition: jobflow.Flow | Job | list[Job] | list[jobflow.Flow] | None = None
     replace: jobflow.Flow | Job | list[Job] | list[jobflow.Flow] | None = None

{jobflow-0.1.9 → jobflow-0.1.11}/src/jobflow/core/maker.py

@@ -149,8 +149,9 @@ class Maker(MSONable):
         update
             The updates to apply.
         name_filter
-            A filter for the Maker name. Only Makers with a matching name will be updated.
-            Includes partial matches, e.g. "ad" will match a Maker with the name "adder".
+            A filter for the Maker name. Only Makers with a matching name will be
+            updated. Includes partial matches, e.g. "ad" will match a Maker with the
+            name "adder".
         class_filter
             A filter for the maker class. Only Makers with a matching class will be
             updated. Note the class filter will match any subclasses.
@@ -232,7 +233,7 @@ class Maker(MSONable):
 
 def recursive_call(
     obj: Maker,
-    func: Callable[[Maker], Any],
+    func: Callable[[Maker], Maker],
     name_filter: str | None = None,
     class_filter: type[Maker] | None = None,
     nested: bool = True,
@@ -281,13 +282,11 @@ def recursive_call(
     # 2. Regenerate the classes and check if they are a Maker
     # 3. Call the functions on the deepest classes first
     # 4. Call the function or update on the deepest classes and move up the tree
-    # 5. Finally replace the call/update the object itself
+    # 5. Finally, replace the call/update the object itself
 
     # find all classes in the serialized maker kwargs
-    if nested:
-        locations = find_key(d, "@class", nested=True)
-    else:
-        locations = [[]]  # will only look at the top level if nested=False
+    # will only look at the top level if nested=False
+    locations = find_key(d, "@class", nested=True) if nested else [[]]
 
     for location in sorted(
         locations, key=len, reverse=True
@@ -298,7 +297,7 @@ def recursive_call(
         if _filter(nested_class):
             # either update or call the function on the nested Maker
             modified_class = func(nested_class)
-            if not isinstance(modified_class, Maker):  # pragma: no cover
+            if not isinstance(modified_class, Maker):
                 raise ValueError(
                     "Function must return a Maker object. "
                     f"Got {type(modified_class)} instead."

{jobflow-0.1.9 → jobflow-0.1.11}/src/jobflow/core/reference.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import contextlib
 import typing
 from typing import Any, Sequence
 
@@ -93,7 +94,7 @@ class OutputReference(MSONable):
     def __init__(
         self,
         uuid: str,
-        attributes: tuple[tuple[str, Any], ...] = tuple(),
+        attributes: tuple[tuple[str, Any], ...] = (),
         output_schema: type[BaseModel] | None = None,
     ):
         super().__init__()
@@ -157,12 +158,10 @@ class OutputReference(MSONable):
         index = None if result is None else result["index"]
 
         if index is not None and index not in cache[self.uuid]:
-            try:
+            with contextlib.suppress(ValueError):
                 cache[self.uuid][index] = store.get_output(
                     self.uuid, which="last", load=True, on_missing=on_missing
                 )
-            except ValueError:
-                pass
 
         if on_missing == OnMissing.ERROR and index not in cache[self.uuid]:
             istr = f" ({index})" if index is not None else ""
@@ -183,12 +182,8 @@ class OutputReference(MSONable):
         cache[self.uuid][index] = data
 
         for attr_type, attr in self.attributes:
-            if attr_type == "i":
-                # index
-                data = data[attr]
-            else:
-                # attribute access
-                data = getattr(data, attr)
+            # i means index else use attribute access
+            data = data[attr] if attr_type == "i" else getattr(data, attr)
 
         return data
 
@@ -226,7 +221,7 @@ class OutputReference(MSONable):
             _, subschema = validate_schema_access(self.output_schema, item)
 
         return OutputReference(
-            self.uuid, self.attributes + (("i", item),), output_schema=subschema
+            self.uuid, (*self.attributes, ("i", item)), output_schema=subschema
         )
 
     def __getattr__(self, item) -> OutputReference:
@@ -242,7 +237,7 @@ class OutputReference(MSONable):
             _, subschema = validate_schema_access(self.output_schema, item)
 
         return OutputReference(
-            self.uuid, self.attributes + (("a", item),), output_schema=subschema
+            self.uuid, (*self.attributes, ("a", item)), output_schema=subschema
         )
 
     def __setattr__(self, attr, val):
@@ -390,13 +385,13 @@ def find_and_get_references(arg: Any) -> tuple[OutputReference, ...]:
 
     if isinstance(arg, OutputReference):
         # if the argument is a reference then stop there
-        return tuple([arg])
+        return (arg,)
 
     elif isinstance(arg, (float, int, str, bool)):
         # argument is a primitive, we won't find a reference here
-        return tuple()
+        return ()
 
-    arg = jsanitize(arg, strict=True, enum_values=True)
+    arg = jsanitize(arg, strict=True, enum_values=True, allow_bson=True)
 
     # recursively find any reference classes
     locations = find_key_value(arg, "@class", "OutputReference")
@@ -453,7 +448,7 @@ def find_and_resolve_references(
         return arg
 
     # serialize the argument to a dictionary
-    encoded_arg = jsanitize(arg, strict=True, enum_values=True)
+    encoded_arg = jsanitize(arg, strict=True, enum_values=True, allow_bson=True)
 
     # recursively find any reference classes
     locations = find_key_value(encoded_arg, "@class", "OutputReference")

{jobflow-0.1.9 → jobflow-0.1.11}/src/jobflow/core/state.py

@@ -24,10 +24,13 @@ __all__ = ["CURRENT_JOB"]
 
 @singleton
 class State:
+    """State of the current job and store."""
+
     job: jobflow.Job | None = None
     store: jobflow.JobStore | None = None
 
     def reset(self):
+        """Reset the current state."""
         self.job = None
         self.store = None