hydra-core 1.3.2__tar.gz → 1.4.0.dev1__tar.gz

{hydra-core-1.3.2/hydra_core.egg-info → hydra_core-1.4.0.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hydra-core
-Version: 1.3.2
+Version: 1.4.0.dev1
 Summary: A framework for elegantly configuring complex applications
 Home-page: https://github.com/facebookresearch/hydra
 Author: Omry Yadan
@@ -9,7 +9,6 @@ License: MIT
 Keywords: command-line configuration yaml tab-completion
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Development Status :: 4 - Beta
-Classifier: Programming Language :: Python :: 3.7
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
@@ -19,8 +18,11 @@ Classifier: Operating System :: MacOS
 Classifier: Operating System :: Microsoft :: Windows
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: omegaconf>=2.4.0.dev2
+Requires-Dist: importlib-resources; python_version < "3.9"
+Requires-Dist: packaging
 
-<p align="center"><img src="https://raw.githubusercontent.com/facebookresearch/hydra/master/website/static/img/Hydra-Readme-logo2.svg" alt="logo" width="70%" /></p>
+<p align="center"><img src="https://raw.githubusercontent.com/facebookresearch/hydra/main/website/static/img/Hydra-Readme-logo2.svg" alt="logo" width="70%" /></p>
 
 <p align="center">
   <a href="https://pypi.org/project/hydra-core/">

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/README.md

@@ -1,4 +1,4 @@
-<p align="center"><img src="https://raw.githubusercontent.com/facebookresearch/hydra/master/website/static/img/Hydra-Readme-logo2.svg" alt="logo" width="70%" /></p>
+<p align="center"><img src="https://raw.githubusercontent.com/facebookresearch/hydra/main/website/static/img/Hydra-Readme-logo2.svg" alt="logo" width="70%" /></p>
 
 <p align="center">
   <a href="https://pypi.org/project/hydra-core/">

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/build_helpers/build_helpers.py

@@ -6,7 +6,9 @@ import os
 import re
 import shutil
 import subprocess
+from functools import partial
 from os.path import abspath, basename, dirname, exists, isdir, join
+from pathlib import Path
 from typing import List, Optional
 
 from setuptools import Command
@@ -185,7 +187,7 @@ class ANTLRCommand(Command):  # type: ignore
             command = [
                 "java",
                 "-jar",
-                join(root_dir, "bin/antlr-4.9.3-complete.jar"),
+                join(root_dir, "bin/antlr-4.11.1-complete.jar"),
                 "-Dlanguage=Python3",
                 "-o",
                 join(project_root, "hydra/grammar/gen/"),
@@ -198,8 +200,37 @@ class ANTLRCommand(Command):  # type: ignore
 
             subprocess.check_call(command)
 
+            log.info("Replacing imports of antlr4 in generated parsers")
+            self._fix_imports()
+
     def initialize_options(self) -> None:
         pass
 
     def finalize_options(self) -> None:
         pass
+
+    def _fix_imports(self) -> None:
+        """Fix imports from the generated parsers to use the vendored antlr4 instead"""
+        build_dir = Path(__file__).parent.absolute()
+        project_root = build_dir.parent
+        lib = "antlr4"
+        pkgname = 'omegaconf.vendor'
+
+        replacements = [
+            partial(  # import antlr4 -> import omegaconf.vendor.antlr4
+                re.compile(r'(^\s*)import {}\n'.format(lib), flags=re.M).sub,
+                r'\1from {} import {}\n'.format(pkgname, lib)
+            ),
+            partial(  # from antlr4 -> from fomegaconf.vendor.antlr4
+                re.compile(r'(^\s*)from {}(\.|\s+)'.format(lib), flags=re.M).sub,
+                r'\1from {}.{}\2'.format(pkgname, lib)
+            ),
+        ]
+
+        path = project_root / "hydra" / "grammar" / "gen"
+        for item in path.iterdir():
+            if item.is_file() and item.name.endswith(".py"):
+                text = item.read_text('utf8')
+                for replacement in replacements:
+                    text = replacement(text)
+                item.write_text(text, 'utf8')

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/build_helpers/test_helpers.py

@@ -112,8 +112,8 @@ def test_find(
         scan_exclude=scan_exclude,
     )
 
-    ret_set = set([str(Path(x)) for x in ret])
-    expected_set = set([str(Path(x)) for x in expected])
+    ret_set = {str(Path(x)) for x in ret}
+    expected_set = {str(Path(x)) for x in expected}
     assert ret_set == expected_set
 
 

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/hydra/__init__.py

@@ -1,7 +1,7 @@
 # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
 
 # Source of truth for Hydra's version
-__version__ = "1.3.2"
+__version__ = "1.4.0.dev1"
 from hydra import utils
 from hydra.errors import MissingConfigException
 from hydra.main import main

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/hydra/_internal/config_repository.py

@@ -27,34 +27,27 @@ from .sources_registry import SourcesRegistry
 
 class IConfigRepository(ABC):
     @abstractmethod
-    def get_schema_source(self) -> ConfigSource:
-        ...
+    def get_schema_source(self) -> ConfigSource: ...
 
     @abstractmethod
-    def load_config(self, config_path: str) -> Optional[ConfigResult]:
-        ...
+    def load_config(self, config_path: str) -> Optional[ConfigResult]: ...
 
     @abstractmethod
-    def group_exists(self, config_path: str) -> bool:
-        ...
+    def group_exists(self, config_path: str) -> bool: ...
 
     @abstractmethod
-    def config_exists(self, config_path: str) -> bool:
-        ...
+    def config_exists(self, config_path: str) -> bool: ...
 
     @abstractmethod
     def get_group_options(
         self, group_name: str, results_filter: Optional[ObjectType] = ObjectType.CONFIG
-    ) -> List[str]:
-        ...
+    ) -> List[str]: ...
 
     @abstractmethod
-    def get_sources(self) -> List[ConfigSource]:
-        ...
+    def get_sources(self) -> List[ConfigSource]: ...
 
     @abstractmethod
-    def initialize_sources(self, config_search_path: ConfigSearchPath) -> None:
-        ...
+    def initialize_sources(self, config_search_path: ConfigSearchPath) -> None: ...
 
 
 class ConfigRepository(IConfigRepository):

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/hydra/_internal/core_plugins/basic_sweeper.py

@@ -62,7 +62,7 @@ class BasicSweeper(Sweeper):
         """
         Instantiates
         """
-        super(BasicSweeper, self).__init__()
+        super().__init__()
 
         if params is None:
             params = {}

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/hydra/_internal/core_plugins/importlib_resources_config_source.py

@@ -30,7 +30,7 @@ class ImportlibResourcesConfigSource(ConfigSource):
 
     def _read_config(self, res: Any) -> ConfigResult:
         try:
-            if sys.version_info[0:2] >= (3, 8) and isinstance(res, zipfile.Path):
+            if isinstance(res, zipfile.Path):
                 # zipfile does not support encoding, read() calls returns bytes.
                 f = res.open()
             else:
@@ -38,8 +38,10 @@ class ImportlibResourcesConfigSource(ConfigSource):
             header_text = f.read(512)
             if isinstance(header_text, bytes):
                 # if header is bytes, utf-8 decode (zipfile path)
-                header_text = header_text.decode("utf-8")
-            header = ConfigSource._get_header_dict(header_text)
+                header_text_str = header_text.decode("utf-8")
+            else:
+                header_text_str = header_text
+            header = ConfigSource._get_header_dict(header_text_str)
             f.seek(0)
             cfg = OmegaConf.load(f)
             return ConfigResult(

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/hydra/_internal/core_plugins/zsh_completion.py

@@ -10,7 +10,7 @@ log = logging.getLogger(__name__)
 
 class ZshCompletion(CompletionPlugin):
     def __init__(self, config_loader: ConfigLoader):
-        super(ZshCompletion, self).__init__(config_loader)
+        super().__init__(config_loader)
         from hydra._internal.core_plugins.bash_completion import BashCompletion
 
         self.delegate = BashCompletion(config_loader)

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/hydra/_internal/defaults_list.py

@@ -312,7 +312,7 @@ def _check_not_missing(
                 group_path,
                 results_filter=ObjectType.CONFIG,
             )
-            opt_list = "\n".join(["\t" + x for x in options])
+            opt_list = "\n".join("\t" + x for x in options)
             msg = dedent(
                 f"""\
                 You must specify '{override_key}', e.g, {override_key}=<OPTION>
@@ -778,7 +778,7 @@ def config_not_found_error(repo: IConfigRepository, tree: DefaultsTreeNode) -> N
         if isinstance(element, GroupDefault):
             msg = f"Could not find '{element.get_config_path()}'\n"
             if options is not None and len(options) > 0:
-                opt_list = "\n".join(["\t" + x for x in options])
+                opt_list = "\n".join("\t" + x for x in options)
                 msg = f"{msg}\nAvailable options in '{group}':\n" + opt_list
         else:
             msg = dedent(

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/hydra/_internal/grammar/grammar_functions.py

@@ -15,7 +15,7 @@ from hydra.core.override_parser.types import (
     Sweep,
 )
 
-ElementType = Union[str, int, bool, float, list, dict]
+ElementType = Union[str, int, bool, float, List[Any], Dict[str, Any]]
 
 
 def apply_to_dict_values(

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/hydra/_internal/hydra.py

@@ -233,7 +233,7 @@ class Hydra:
 
         for shell, plugins in shell_to_plugin.items():
             if len(plugins) > 1:
-                lst = ",".join([type(plugin).__name__ for plugin in plugins])
+                lst = ",".join(type(plugin).__name__ for plugin in plugins)
                 raise ValueError(f"Multiple plugins installed for {shell} : {lst}")
 
         return shell_to_plugin
@@ -251,7 +251,7 @@ class Hydra:
 
         def find_plugin(cmd: str) -> CompletionPlugin:
             if cmd not in shell_to_plugin:
-                lst = "\n".join(["\t" + x for x in shell_to_plugin.keys()])
+                lst = "\n".join("\t" + x for x in shell_to_plugin.keys())
                 raise ValueError(
                     f"No completion plugin for '{cmd}' found, available : \n{lst}"
                 )
@@ -287,7 +287,7 @@ class Hydra:
             if parent == "":
                 group_name = group
             else:
-                group_name = "{}/{}".format(parent, group)
+                group_name = f"{parent}/{group}"
             files = self.config_loader.get_group_options(group_name, ObjectType.CONFIG)
             dirs = self.config_loader.get_group_options(group_name, ObjectType.GROUP)
             if len(files) > 0:
@@ -305,10 +305,10 @@ class Hydra:
             options = sorted(self.config_loader.get_group_options(group))
             if compact:
                 items = ", ".join(options)
-                line = "{}: {}".format(group, items)
+                line = f"{group}: {items}"
             else:
-                items = "\n".join(["  " + o for o in options])
-                line = "{}:\n{}".format(group, items)
+                items = "\n".join("  " + o for o in options)
+                line = f"{group}:\n{items}"
             s += line + "\n"
 
         return s
@@ -396,14 +396,14 @@ class Hydra:
             if len(plugins) > 0:
                 Hydra._log_header(header=f"{plugin_type.__name__}:", prefix="\t")
                 for plugin in plugins:
-                    log.debug("\t\t{}".format(plugin.__name__))
+                    log.debug(f"\t\t{plugin.__name__}")
                     if plugin.__name__ in all_plugins:
                         all_plugins.remove(plugin.__name__)
 
         if len(all_plugins) > 0:
             Hydra._log_header(header="Generic plugins: ", prefix="\t")
             for plugin_name in all_plugins:
-                log.debug("\t\t{}".format(plugin_name))
+                log.debug(f"\t\t{plugin_name}")
 
     def _print_search_path(
         self,

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/hydra/_internal/utils.py

@@ -494,7 +494,7 @@ def _get_completion_help() -> str:
             completion_info.append(plugin_cls.help(cmd).format(_get_exec_command()))
         completion_info.append("")
 
-    completion_help = "\n".join([f"    {x}" if x else x for x in completion_info])
+    completion_help = "\n".join(f"    {x}" if x else x for x in completion_info)
     return completion_help
 
 

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/hydra/core/config_loader.py

@@ -23,26 +23,21 @@ class ConfigLoader(ABC):
         run_mode: RunMode,
         from_shell: bool = True,
         validate_sweep_overrides: bool = True,
-    ) -> DictConfig:
-        ...
+    ) -> DictConfig: ...
 
     @abstractmethod
     def load_sweep_config(
         self, master_config: DictConfig, sweep_overrides: List[str]
-    ) -> DictConfig:
-        ...
+    ) -> DictConfig: ...
 
     @abstractmethod
-    def get_search_path(self) -> ConfigSearchPath:
-        ...
+    def get_search_path(self) -> ConfigSearchPath: ...
 
     @abstractmethod
-    def get_sources(self) -> List[ConfigSource]:
-        ...
+    def get_sources(self) -> List[ConfigSource]: ...
 
     @abstractmethod
-    def list_groups(self, parent_name: str) -> List[str]:
-        ...
+    def list_groups(self, parent_name: str) -> List[str]: ...
 
     @abstractmethod
     def get_group_options(
@@ -51,8 +46,7 @@ class ConfigLoader(ABC):
         results_filter: Optional[ObjectType] = ObjectType.CONFIG,
         config_name: Optional[str] = None,
         overrides: Optional[List[str]] = None,
-    ) -> List[str]:
-        ...
+    ) -> List[str]: ...
 
     @abstractmethod
     def compute_defaults_list(
@@ -60,5 +54,4 @@ class ConfigLoader(ABC):
         config_name: Optional[str],
         overrides: List[str],
         run_mode: RunMode,
-    ) -> Any:
-        ...
+    ) -> Any: ...

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/hydra/core/config_search_path.py

@@ -28,8 +28,7 @@ class SearchPathQuery:
 
 class ConfigSearchPath(ABC):
     @abstractmethod
-    def get_path(self) -> MutableSequence[SearchPathElement]:
-        ...
+    def get_path(self) -> MutableSequence[SearchPathElement]: ...
 
     @abstractmethod
     def append(

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/hydra/core/config_store.py

@@ -28,8 +28,7 @@ class ConfigStoreWithProvider:
             group=group, name=name, node=node, package=package, provider=self.provider
         )
 
-    def __exit__(self, exc_type: Any, exc_value: Any, exc_traceback: Any) -> Any:
-        ...
+    def __exit__(self, exc_type: Any, exc_value: Any, exc_traceback: Any) -> Any: ...
 
 
 @dataclass
@@ -133,10 +132,10 @@ class ConfigStore(metaclass=Singleton):
     def list(self, path: str) -> List[str]:
         d = self._open(path)
         if d is None:
-            raise IOError(f"Path not found {path}")
+            raise OSError(f"Path not found {path}")
 
         if not isinstance(d, dict):
-            raise IOError(f"Path points to a file : {path}")
+            raise OSError(f"Path points to a file : {path}")
 
         return sorted(d.keys())
 

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/hydra/core/override_parser/overrides_parser.py

@@ -2,7 +2,10 @@
 import sys
 from typing import Any, List, Optional
 
-from antlr4.error.Errors import LexerNoViableAltException, RecognitionException
+from omegaconf.vendor.antlr4.error.Errors import (
+    LexerNoViableAltException,
+    RecognitionException,
+)
 
 from hydra._internal.grammar import grammar_functions
 from hydra._internal.grammar.functions import Functions

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/hydra/core/override_parser/overrides_visitor.py

@@ -3,9 +3,13 @@ import sys
 import warnings
 from typing import Any, Dict, List, Optional, Tuple, Union
 
-from antlr4 import ParserRuleContext, TerminalNode, Token
-from antlr4.error.ErrorListener import ErrorListener
-from antlr4.tree.Tree import TerminalNodeImpl
+from omegaconf.vendor.antlr4 import (  # type: ignore[attr-defined]
+    ParserRuleContext,
+    TerminalNode,
+    Token,
+)
+from omegaconf.vendor.antlr4.error.ErrorListener import ErrorListener
+from omegaconf.vendor.antlr4.tree.Tree import TerminalNodeImpl
 
 from hydra._internal.grammar.functions import FunctionCall, Functions
 from hydra._internal.grammar.utils import _ESC_QUOTED_STR
@@ -107,7 +111,8 @@ class HydraOverrideVisitor(OverrideParserVisitor):
         item = next(children)
         assert isinstance(item, OverrideParser.DictKeyContext)
         pkey = self.visitDictKey(item)
-        assert self.is_matching_terminal(next(children), OverrideLexer.COLON)
+        colon = next(children)
+        assert self.is_matching_terminal(colon, OverrideLexer.COLON)
         value = next(children)
         assert isinstance(value, OverrideParser.ElementContext)
         return pkey, self.visitElement(value)
@@ -161,7 +166,7 @@ class HydraOverrideVisitor(OverrideParserVisitor):
         if (
             override_type == OverrideType.DEL
             and isinstance(eq_node, TerminalNode)
-            and eq_node.symbol.type == Token.EOF
+            and eq_node.symbol.type == Token.EOF  # type: ignore[attr-defined]
         ):
             value = None
             value_type = None
@@ -212,7 +217,8 @@ class HydraOverrideVisitor(OverrideParserVisitor):
         kwargs = {}
         children = ctx.getChildren()
         func_name = next(children).getText()
-        assert self.is_matching_terminal(next(children), OverrideLexer.POPEN)
+        popen = next(children)
+        assert self.is_matching_terminal(popen, OverrideLexer.POPEN)
         in_kwargs = False
         while True:
             cur = next(children)
@@ -242,26 +248,26 @@ class HydraOverrideVisitor(OverrideParserVisitor):
             ) from e
 
     def _createPrimitive(
-        self, ctx: ParserRuleContext
+        self, ctx: ParserRuleContext  # type: ignore[valid-type]
     ) -> Optional[Union[QuotedString, int, bool, float, str]]:
         ret: Optional[Union[int, bool, float, str]]
         first_idx = 0
-        last_idx = ctx.getChildCount()
+        last_idx = ctx.getChildCount()  # type: ignore[attr-defined]
         # skip first if whitespace
-        if self.is_ws(ctx.getChild(0)):
+        if self.is_ws(ctx.getChild(0)):  # type: ignore[attr-defined]
             if last_idx == 1:
                 # Only whitespaces => this is not allowed.
                 raise HydraException(
                     "Trying to parse a primitive that is all whitespaces"
                 )
             first_idx = 1
-        if self.is_ws(ctx.getChild(-1)):
+        if self.is_ws(ctx.getChild(-1)):  # type: ignore[attr-defined]
             last_idx = last_idx - 1
         num = last_idx - first_idx
         if num > 1:
             # Concatenate, while un-escaping as needed.
             tokens = []
-            for i, n in enumerate(ctx.getChildren()):
+            for i, n in enumerate(ctx.getChildren()):  # type: ignore[attr-defined]
                 if n.symbol.type == OverrideLexer.WS and (
                     i < first_idx or i >= last_idx
                 ):
@@ -274,7 +280,7 @@ class HydraOverrideVisitor(OverrideParserVisitor):
                 )
             ret = "".join(tokens)
         else:
-            node = ctx.getChild(first_idx)
+            node = ctx.getChild(first_idx)  # type: ignore[attr-defined]
             if node.symbol.type == OverrideLexer.QUOTED_VALUE:
                 text = node.getText()
                 qc = text[0]
@@ -358,7 +364,7 @@ class HydraOverrideVisitor(OverrideParserVisitor):
         return "".join(tokens)
 
 
-class HydraErrorListener(ErrorListener):  # type: ignore
+class HydraErrorListener(ErrorListener):
     def syntaxError(
         self,
         recognizer: Any,

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/hydra/core/plugins.py

@@ -187,13 +187,14 @@ class Plugins(metaclass=Singleton):
                             assert m is not None
                             loaded_mod = m.load_module(modname)
                         else:
-                            spec = importer.find_spec(modname)
+                            spec = importer.find_spec(modname)  # type: ignore[call-arg]
                             assert spec is not None
                             if modname in sys.modules:
                                 loaded_mod = sys.modules[modname]
                             else:
                                 loaded_mod = importlib.util.module_from_spec(spec)
                             if loaded_mod is not None:
+                                assert spec.loader is not None
                                 spec.loader.exec_module(loaded_mod)
                                 sys.modules[modname] = loaded_mod
 

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/hydra/core/singleton.py

@@ -10,7 +10,7 @@ class Singleton(type):
 
     def __call__(cls, *args: Any, **kwargs: Any) -> Any:
         if cls not in cls._instances:
-            cls._instances[cls] = super(Singleton, cls).__call__(*args, **kwargs)
+            cls._instances[cls] = super().__call__(*args, **kwargs)
         return cls._instances[cls]
 
     def instance(cls: Any, *args: Any, **kwargs: Any) -> Any:

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/hydra/errors.py

@@ -2,31 +2,26 @@
 from typing import Optional, Sequence
 
 
-class HydraException(Exception):
-    ...
+class HydraException(Exception): ...
 
 
-class CompactHydraException(HydraException):
-    ...
+class CompactHydraException(HydraException): ...
 
 
 class OverrideParseException(CompactHydraException):
     def __init__(self, override: str, message: str) -> None:
-        super(OverrideParseException, self).__init__(message)
+        super().__init__(message)
         self.override = override
         self.message = message
 
 
-class InstantiationException(CompactHydraException):
-    ...
+class InstantiationException(CompactHydraException): ...
 
 
-class ConfigCompositionException(CompactHydraException):
-    ...
+class ConfigCompositionException(CompactHydraException): ...
 
 
-class SearchPathException(CompactHydraException):
-    ...
+class SearchPathException(CompactHydraException): ...
 
 
 class MissingConfigException(IOError, ConfigCompositionException):
@@ -36,10 +31,9 @@ class MissingConfigException(IOError, ConfigCompositionException):
         missing_cfg_file: Optional[str] = None,
         options: Optional[Sequence[str]] = None,
     ) -> None:
-        super(MissingConfigException, self).__init__(message)
+        super().__init__(message)
         self.missing_cfg_file = missing_cfg_file
         self.options = options
 
 
-class HydraDeprecationError(HydraException):
-    ...
+class HydraDeprecationError(HydraException): ...

{hydra-core-1.3.2 → hydra_core-1.4.0.dev1}/hydra/experimental/initialize.py

@@ -46,6 +46,7 @@ class initialize:
             config_path=config_path,
             job_name=job_name,
             caller_stack_depth=caller_stack_depth + 1,
+            version_base=str(version.getbase()),
         )
 
     def __enter__(self, *args: Any, **kwargs: Any) -> None:
@@ -80,7 +81,9 @@ class initialize_config_module:
         deprecation_warning(message=message)
 
         self.delegate = real_initialize_config_module(
-            config_module=config_module, job_name=job_name
+            config_module=config_module,
+            job_name=job_name,
+            version_base=str(version.getbase()),
         )
 
     def __enter__(self, *args: Any, **kwargs: Any) -> None:
@@ -116,7 +119,9 @@ class initialize_config_dir:
         deprecation_warning(message=message)
 
         self.delegate = real_initialize_config_dir(
-            config_dir=config_dir, job_name=job_name
+            config_dir=config_dir,
+            job_name=job_name,
+            version_base=str(version.getbase()),
         )
 
     def __enter__(self, *args: Any, **kwargs: Any) -> None: