PyPI - experimaestro - Versions diffs - 1.6.1__py3-none-any.whl → 1.15.2__py3-none-any.whl - Mend

experimaestro 1.6.1py3-none-any.whl → 1.15.2py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (98) hide show

experimaestro/__init__.py +14 -3
experimaestro/annotations.py +13 -3
experimaestro/cli/filter.py +19 -5
experimaestro/cli/jobs.py +12 -5
experimaestro/commandline.py +3 -7
experimaestro/connectors/__init__.py +27 -12
experimaestro/connectors/local.py +19 -10
experimaestro/connectors/ssh.py +1 -1
experimaestro/core/arguments.py +35 -3
experimaestro/core/callbacks.py +52 -0
experimaestro/core/context.py +8 -9
experimaestro/core/identifier.py +301 -0
experimaestro/core/objects/__init__.py +44 -0
experimaestro/core/{objects.py → objects/config.py} +364 -716
experimaestro/core/objects/config_utils.py +58 -0
experimaestro/core/objects/config_walk.py +151 -0
experimaestro/core/objects.pyi +15 -45
experimaestro/core/serialization.py +63 -9
experimaestro/core/serializers.py +1 -8
experimaestro/core/types.py +61 -6
experimaestro/experiments/cli.py +79 -29
experimaestro/experiments/configuration.py +3 -0
experimaestro/generators.py +6 -1
experimaestro/ipc.py +4 -1
experimaestro/launcherfinder/parser.py +8 -3
experimaestro/launcherfinder/registry.py +29 -10
experimaestro/launcherfinder/specs.py +49 -10
experimaestro/launchers/slurm/base.py +51 -13
experimaestro/mkdocs/__init__.py +1 -1
experimaestro/notifications.py +2 -1
experimaestro/run.py +3 -1
experimaestro/scheduler/base.py +114 -6
experimaestro/scheduler/dynamic_outputs.py +184 -0
experimaestro/scheduler/state.py +75 -0
experimaestro/scheduler/workspace.py +2 -1
experimaestro/scriptbuilder.py +13 -2
experimaestro/server/data/0c35d18bf06992036b69.woff2 +0 -0
experimaestro/server/data/1815e00441357e01619e.ttf +0 -0
experimaestro/server/data/219aa9140e099e6c72ed.woff2 +0 -0
experimaestro/server/data/2463b90d9a316e4e5294.woff2 +0 -0
experimaestro/server/data/2582b0e4bcf85eceead0.ttf +0 -0
experimaestro/server/data/3a4004a46a653d4b2166.woff +0 -0
experimaestro/server/data/3baa5b8f3469222b822d.woff +0 -0
experimaestro/server/data/4d73cb90e394b34b7670.woff +0 -0
experimaestro/server/data/4ef4218c522f1eb6b5b1.woff2 +0 -0
experimaestro/server/data/5d681e2edae8c60630db.woff +0 -0
experimaestro/server/data/6f420cf17cc0d7676fad.woff2 +0 -0
experimaestro/server/data/89999bdf5d835c012025.woff2 +0 -0
experimaestro/server/data/914997e1bdfc990d0897.ttf +0 -0
experimaestro/server/data/c210719e60948b211a12.woff2 +0 -0
experimaestro/server/data/c380809fd3677d7d6903.woff2 +0 -0
experimaestro/server/data/f882956fd323fd322f31.woff +0 -0
experimaestro/server/data/favicon.ico +0 -0
experimaestro/server/data/index.css +22963 -0
experimaestro/server/data/index.css.map +1 -0
experimaestro/server/data/index.html +27 -0
experimaestro/server/data/index.js +101770 -0
experimaestro/server/data/index.js.map +1 -0
experimaestro/server/data/login.html +22 -0
experimaestro/server/data/manifest.json +15 -0
experimaestro/settings.py +2 -2
experimaestro/sphinx/__init__.py +7 -17
experimaestro/taskglobals.py +7 -2
experimaestro/tests/core/__init__.py +0 -0
experimaestro/tests/core/test_generics.py +206 -0
experimaestro/tests/definitions_types.py +5 -3
experimaestro/tests/launchers/bin/sbatch +34 -7
experimaestro/tests/launchers/bin/srun +5 -0
experimaestro/tests/launchers/common.py +16 -4
experimaestro/tests/restart.py +9 -4
experimaestro/tests/tasks/all.py +23 -10
experimaestro/tests/tasks/foreign.py +2 -4
experimaestro/tests/test_dependencies.py +0 -6
experimaestro/tests/test_experiment.py +73 -0
experimaestro/tests/test_findlauncher.py +11 -4
experimaestro/tests/test_forward.py +5 -5
experimaestro/tests/test_generators.py +93 -0
experimaestro/tests/test_identifier.py +114 -99
experimaestro/tests/test_instance.py +6 -21
experimaestro/tests/test_objects.py +20 -4
experimaestro/tests/test_param.py +60 -22
experimaestro/tests/test_serializers.py +24 -64
experimaestro/tests/test_tags.py +5 -11
experimaestro/tests/test_tasks.py +10 -23
experimaestro/tests/test_tokens.py +3 -2
experimaestro/tests/test_types.py +20 -17
experimaestro/tests/test_validation.py +48 -91
experimaestro/tokens.py +16 -5
experimaestro/typingutils.py +8 -8
experimaestro/utils/asyncio.py +6 -2
experimaestro/utils/multiprocessing.py +44 -0
experimaestro/utils/resources.py +7 -3
{experimaestro-1.6.1.dist-info → experimaestro-1.15.2.dist-info}/METADATA +27 -34
experimaestro-1.15.2.dist-info/RECORD +159 -0
{experimaestro-1.6.1.dist-info → experimaestro-1.15.2.dist-info}/WHEEL +1 -1
experimaestro-1.6.1.dist-info/RECORD +0 -122
{experimaestro-1.6.1.dist-info → experimaestro-1.15.2.dist-info}/entry_points.txt +0 -0
{experimaestro-1.6.1.dist-info → experimaestro-1.15.2.dist-info/licenses}/LICENSE +0 -0

experimaestro/server/data/login.html ADDED Viewed

@@ -0,0 +1,22 @@
+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <meta charset="utf-8" />
+    <meta
+      name="viewport"
+      content="width=device-width, initial-scale=1, shrink-to-fit=no"
+    />
+    <meta name="theme-color" content="#000000" />
+    <title>Experimaestro</title>
+  </head>
+  <body>
+    <h1>Experimaestro</h1>
+    <form action="/auth" method="GET">
+      <div>
+        Token
+        <input type="text" name="xpm-token" />
+        <input type="submit" value="submit" />
+      </div>
+    </form>
+  </body>
+</html>

experimaestro/server/data/manifest.json ADDED Viewed

@@ -0,0 +1,15 @@
+{
+  "short_name": "Experimaestro",
+  "name": "Experimaestro Client",
+  "icons": [
+    {
+      "src": "favicon.ico",
+      "sizes": "64x64 32x32 24x24 16x16",
+      "type": "image/x-icon"
+    }
+  ],
+  "start_url": "./index.html",
+  "display": "standalone",
+  "theme_color": "#000000",
+  "background_color": "#ffffff"
+}

experimaestro/settings.py CHANGED Viewed

@@ -37,7 +37,7 @@ class WorkspaceSettings:
     alt_workspaces: List[str] = field(default_factory=list)
     """Alternative workspaces to find jobs or experiments"""
     def __post_init__(self):
         self.path = self.path.expanduser().resolve()
@@ -83,7 +83,7 @@ def get_workspace(id: Optional[str] = None) -> Optional[WorkspaceSettings]:
     return None
-def find_workspace(*, workspace: Optional[str] = None, workdir: Optional[Path] = None):
+def find_workspace(*, workspace: Optional[str] = None, workdir: Optional[Path] = None) -> WorkspaceSettings:
     """Find workspace"""
     workdir = Path(workdir) if workdir else None

experimaestro/sphinx/__init__.py CHANGED Viewed

@@ -10,19 +10,13 @@ from docutils import nodes
 from sphinx.application import Sphinx
 from sphinx import addnodes
 from sphinx.ext.autodoc import ClassDocumenter, Documenter, restify
-from sphinx.locale import _, __
+from sphinx.locale import _
 from sphinx.util import inspect, logging
-from sphinx.domains.python import (
-    PyClasslike,
-    PyAttribute,
-    PyObject,
-    directives,
-    desc_signature,
-    _parse_annotation,
-)
+from sphinx.domains.python import PyClasslike, PyAttribute, directives
+from sphinx.domains.python import PyObject  # noqa: F401
+from sphinx.addnodes import desc_signature
 from sphinx.util.typing import OptionSpec
 from docutils.statemachine import StringList
-import logging
 import re
 from experimaestro import Config, Task
@@ -97,9 +91,6 @@ class ConfigDocumenter(ClassDocumenter):
         can_document = inspect.isclass(member) and issubclass(member, Config)
         return can_document
-    def add_directive_header(self, sig: str) -> None:
-        super().add_directive_header(sig)
     def get_object_members(self, want_all: bool):  # -> Tuple[bool, ObjectMembers]:
         r = super().get_object_members(want_all)
         return r
@@ -157,7 +148,7 @@ class ConfigDocumenter(ClassDocumenter):
         # Our specific code
         if issubclass(self.object, Task):
-            self.add_line(f"   :task:", sourcename)
+            self.add_line("   :task:", sourcename)
         # add inheritance info, if wanted
         if not self.doc_as_attr and self.options.show_inheritance:
@@ -196,7 +187,6 @@ class ConfigDocumenter(ClassDocumenter):
     def add_content(
         self, more_content: Optional[StringList], no_docstring: bool = False
     ) -> None:
         xpminfo = getxpminfo(self.object)
         source_name = self.get_sourcename()
@@ -214,9 +204,9 @@ class ConfigDocumenter(ClassDocumenter):
                     source_name,
                 )
             if argument.generator:
-                self.add_line(f"   :generated:", source_name)
+                self.add_line("   :generated:", source_name)
             elif argument.constant:
-                self.add_line(f"   :constant:", source_name)
+                self.add_line("   :constant:", source_name)
             # self.add_line("", source_name)
             if argument.help:

experimaestro/taskglobals.py CHANGED Viewed

@@ -1,7 +1,6 @@
+from functools import cached_property
 from pathlib import Path
 from typing import Optional
-import os
-import logging
 class Env:
@@ -18,6 +17,12 @@ class Env:
     # - no progress report
     slave: bool = False
+    @cached_property
+    def xpm_path(self):
+        path = self.taskpath / ".experimaestro"
+        path.mkdir(exist_ok=True)
+        return path
     @staticmethod
     def instance():
         if Env._instance is None:

experimaestro/tests/core/__init__.py ADDED Viewed

File without changes

experimaestro/tests/core/test_generics.py ADDED Viewed

@@ -0,0 +1,206 @@
+"""Tests for the use of generics in configurations"""
+from typing import Generic, Optional, TypeVar
+import pytest
+from experimaestro import Config, Param
+from experimaestro.core.arguments import Argument
+from experimaestro.core.types import TypeVarType
+T = TypeVar("T")
+class SimpleConfig(Config):
+    pass
+class SimpleConfigChild(SimpleConfig):
+    pass
+class SimpleGenericConfig(Config, Generic[T]):
+    x: Param[T]
+class SimpleGenericConfigChild(SimpleGenericConfig, Generic[T]):
+    """A child class of SimpleGenericConfig that also uses generics"""
+    pass
+def test_core_generics_typevar():
+    a = SimpleGenericConfig.C(x=1)
+    x_arg = a.__xpmtype__.arguments["x"]
+    # Check correct interpretation of typevar
+    assert type(x_arg) is Argument
+    assert isinstance(x_arg.type, TypeVarType)
+    assert x_arg.type.typevar == T
+    assert isinstance(a.x, int)
+def test_core_generics_simple():
+    a = SimpleGenericConfig.C(x=2)
+    # OK
+    a.x = 3
+    # Fails: changing generics is not allowed
+    with pytest.raises(TypeError):
+        a.x = "a string"
+    # typevar bindings are local to the instance,
+    # so we can create a new instance with a different type
+    SimpleGenericConfig.C(x="a string")
+class DoubleGenericConfig(Config, Generic[T]):
+    x: Param[T]
+    y: Param[T]
+def test_core_generics_double():
+    # OK
+    DoubleGenericConfig.C(x=1, y=1)
+    # Fails
+    with pytest.raises(TypeError):
+        DoubleGenericConfig.C(x=1, y="a")
+    a = DoubleGenericConfig.C(x=1, y=1)
+    a.y = 2
+    with pytest.raises(TypeError):
+        a.x = "b"
+def test_core_generics_double_rebind():
+    a = DoubleGenericConfig.C(x=1, y=1)
+    # Rebinding to a different type should not work
+    with pytest.raises(TypeError):
+        a.x, a.y = "some", "string"
+def test_core_generics_double_plus():
+    # Testing with inheritance
+    # We allow subclasses of the typevar binding
+    # We also allow generalizing up the typevar binding
+    # This means that we can use a super class of the typevar binding
+    # Works
+    a = DoubleGenericConfig.C(x=SimpleConfigChild.C())
+    a.y = SimpleConfig.C()
+    # Works also
+    b = DoubleGenericConfig.C(x=SimpleConfig.C())
+    b.y = SimpleConfigChild.C()
+    a.x = SimpleConfigChild.C()
+    with pytest.raises(TypeError):
+        a.x = "a string"
+def test_core_generics_double_type_escalation():
+    a = DoubleGenericConfig.C(x=SimpleConfigChild.C())
+    a.y = SimpleConfigChild.C()
+    # T is now bound to SimpleConfigChild
+    a.y = SimpleConfig.C()
+    # T is now bound to SimpleConfig
+    a.y = object()
+    # T is now bound to object, which is a super class of SimpleConfigChild
+    # This is allowed, since we are not changing the typevar binding
+    a.x = "a string"
+    a.y = dict()
+    # This is allowed, since we are not changing the typevar binding
+def test_core_generics_double_deep_bind():
+    # Since we are deep binding the typevar T to a specific type,
+    # we should not be able to have coherent *local-only* type bindings
+    # The type bindings are transient
+    with pytest.raises(TypeError):
+        DoubleGenericConfig.C(
+            x=DoubleGenericConfig.C(x=1, y=2), y=DoubleGenericConfig.C(x=3, y=4)
+        )
+class NestedConfig(Config, Generic[T]):
+    x: Param[DoubleGenericConfig[T]]
+    y: Param[SimpleGenericConfig[T]]
+def test_core_generics_nested():
+    # OK
+    NestedConfig.C(x=DoubleGenericConfig.C(x=1, y=1), y=SimpleGenericConfig.C(x=2))
+    # Not OK
+    with pytest.raises(TypeError):
+        NestedConfig.C(
+            x=DoubleGenericConfig.C(x=1, y=1), y=SimpleGenericConfig.C(x="b")
+        )
+    with pytest.raises(TypeError):
+        a = NestedConfig.C(
+            x=DoubleGenericConfig.C(x=1, y=1), y=SimpleGenericConfig.C(x=1)
+        )
+        a.x.x = "a string"
+class TreeGenericConfig(Config, Generic[T]):
+    x: Param[T]
+    left: Optional["TreeGenericConfig[T]"] = None
+    right: Optional["TreeGenericConfig[T]"] = None
+class TagTreeGenericConfig(TreeGenericConfig[T], Generic[T]):
+    """A tagged version of TreeGenericConfig to test recursive generics"""
+    tag: Param[str] = "default"
+def test_core_generics_recursive():
+    a = TreeGenericConfig.C(x=1)
+    a.left = TreeGenericConfig.C(x=2)
+    a.right = TreeGenericConfig.C(x=3)
+    with pytest.raises(TypeError):
+        a.left.x = "a string"
+    # OK to use a child class
+    a.left = TagTreeGenericConfig.C(x=4, tag="left")
+    with pytest.raises(TypeError):
+        a.left.x = "a string"
+def test_core_generics_recursive_child():
+    # Testing with a child class on the generic value
+    a = TreeGenericConfig.C(x=SimpleConfig.C())
+    a.left = TreeGenericConfig.C(x=SimpleConfig.C())
+    a.right = TreeGenericConfig.C(x=SimpleConfig.C())
+    a.left.x = SimpleConfigChild.C()
+    with pytest.raises(TypeError):
+        a.left.x = "a string"
+U = TypeVar("U", bound=SimpleConfigChild)
+class BoundGenericConfig(Config, Generic[U]):
+    x: Param[U]
+def test_core_generics_bound_typevar():
+    a = BoundGenericConfig.C(x=SimpleConfigChild.C())
+    assert isinstance(a.x, SimpleConfigChild)
+    with pytest.raises(TypeError):
+        a.x = SimpleConfig.C()

experimaestro/tests/definitions_types.py CHANGED Viewed

@@ -1,15 +1,17 @@
-from experimaestro import argument, Task
+from experimaestro import Param, Task
-@argument("value", type=int)
 class IntegerTask(Task):
+    value: Param[int]
     def execute(self):
         if not isinstance(self.value, int):
             raise AssertionError("Not an integer")
-@argument("value", type=float)
 class FloatTask(Task):
+    value: Param[float]
     def execute(self):
         if not isinstance(self.value, float):
             raise AssertionError("Not a float but %s" % type(self.value))

experimaestro/tests/launchers/bin/sbatch CHANGED Viewed

@@ -2,15 +2,28 @@
 # slurm test suite
+pwd 1>&2
+CURDIR="$(realpath "$(dirname "$0")"/..)"
 # Where we store the jobs
-XPM_SLURM_DIR="$(realpath "$(dirname "$0")"/..)/slurm"
+XPM_SLURM_DIR="${CURDIR}/slurm"
 if ! test -d "$XPM_SLURM_DIR"; then
     echo "Directory $XPM_SLURM_DIR does not exist" 1>&2
     exit 1
 fi
 mkdir -p "$XPM_SLURM_DIR/jobs"
-echo "Slurm directory: $XPM_SLURM_DIR" >&2
+echo "Slurm directory: $XPM_SLURM_DIR" >&2
+RED='\033[0;31m'
+NC='\033[0m' # No Color
+if ! which flock > /dev/null; then
+    echo -e "[${RED}ERROR${NC}] flock is not installed on this system" 1>&2
+    exit 1
+fi
 lockpath() {
     fid="$1"
@@ -18,7 +31,7 @@ lockpath() {
     echo "Locking $path..." 1>&2
     eval exec "$fid<>" $path
-    if ! flock --timeout 2 $fid; then
+    if ! flock --timeout 2 $fid; then
         echo Could not lock "$path" - stopping 1>&2
         exit 017
     fi
@@ -37,12 +50,26 @@ while true; do
         -o) shift; stdout="$1"; shift;;
         -e) shift; stderr="$1"; shift;;
         --*) args+=("$1"); shift;;
-        *) break 2;;
+        *) break 2;;
     esac
 done
-echo "Starting $@ ${args[@]} > $stdout" >&2
-(eval "$@" "${args[@]}"; echo $? > "$XPM_SLURM_DIR/jobs/$$.status") > $stdout 2> $stderr &
+chdir=$(pwd)
+while IFS= read -r line; do
+    case "$line" in
+        "#SBATCH --output="*)stdout=${line#*#SBATCH --output=};;
+        "#SBATCH --error="*) stderr=${line#*#SBATCH --error=};;
+        "#SBATCH --chdir="*) chdir=${line#*#SBATCH --chdir=};;
+    esac
+done < "$1"
+cd "$chdir"
+echo "Starting $@ ${args[@]} > $stdout 2> $stderr" >&2
+(
+    export PATH="${CURDIR}/bin:$PATH"
+    eval "$@" "${args[@]}"
+    echo $? > "$XPM_SLURM_DIR/jobs/$$.status"
+) > $stdout 2> $stderr &
 JOBID="$$"
 date > "$XPM_SLURM_DIR/jobs/$JOBID.start"
 disown
@@ -51,4 +78,4 @@ if test "$parsable" == 0; then
     echo "Submitted batch job ${JOBID}"
 else
     echo "${JOBID};cluster"
-fi
+fi

experimaestro/tests/launchers/bin/srun ADDED Viewed

@@ -0,0 +1,5 @@
+#!/bin/bash
+# Just run the command
+eval "$@"

experimaestro/tests/launchers/common.py CHANGED Viewed

@@ -68,24 +68,36 @@ class WaitUntilTouched(Task):
 def takeback(launcher, datapath, txp1, txp2):
+    """Launch two times the same task (with two experiments)
+    :param launcher: The launcher
+    :param datapath: The path containing the two files that control the task, namely (1) touching which is created by the task when starting, (2) waiting which is controlled here
+    :param txp1: The first experiment
+    :param txp2: The second experiment
+    """
     datapath.mkdir()
     touching = datapath / "touching"
     waiting = datapath / "waiting"
-    with txp1 as xp1:  # flake8: noqa: F841
-        WaitUntilTouched(touching=touching, waiting=waiting).submit(launcher=launcher)
+    with txp1:
+        task: WaitUntilTouched = WaitUntilTouched(
+            touching=touching, waiting=waiting
+        ).submit(launcher=launcher)
         logger.debug("Waiting for task to create 'touching' file")
         while not touching.is_file():
+            if task.__xpm__.job.state.finished():
+                raise Exception("Job has finished... too early")
             time.sleep(0.01)
-        with txp2 as xp2:  # flake8: noqa: F841
+        with txp2:
             result = WaitUntilTouched(touching=touching, waiting=waiting).submit(
                 launcher=launcher
             )
-            logger.debug("Waiting for job to be in running mode")
+            logger.debug("Waiting for job to be running (scheduler)")
             while result.__xpm__.job.state != JobState.RUNNING:
                 time.sleep(0.1)
+            logger.debug("OK, no we can notify the task")
             waiting.touch()

experimaestro/tests/restart.py CHANGED Viewed

@@ -1,12 +1,14 @@
 import time
+from pathlib import Path
 import sys
 from typing import Callable
-from experimaestro import Task, pathoption
+from experimaestro import Task, Meta, field, PathGenerator
 import psutil
 import logging
 import subprocess
 import json
 import signal
 from experimaestro.scheduler.workspace import RunMode
 from experimaestro.tests.utils import TemporaryExperiment, is_posix
 from experimaestro.scheduler import JobState
@@ -25,10 +27,13 @@ TERMINATES_FUNC = [terminate]
 if is_posix():
     TERMINATES_FUNC.append(sigint)
+MAX_RESTART_WAIT = 50  # 5 seconds
-@pathoption("touch", "touch")
-@pathoption("wait", "wait")
 class Restart(Task):
+    touch: Meta[Path] = field(default_factory=PathGenerator("touch"))
+    wait: Meta[Path] = field(default_factory=PathGenerator("wait"))
     def execute(self):
         # Write the file "touch" to notify that we started
         with open(self.touch, "w") as out:
@@ -78,7 +83,7 @@ def restart(terminate: Callable, experiment):
         while not task.touch.is_file():
             time.sleep(0.1)
             counter += 1
-            if counter >= 20:
+            if counter >= MAX_RESTART_WAIT:
                 terminate(xpmprocess)
                 assert False, "Timeout waiting for task to be executed"

experimaestro/tests/tasks/all.py CHANGED Viewed

@@ -1,11 +1,13 @@
+from pathlib import Path
 import time
 from typing import List
 from experimaestro import (
-    param,
+    Meta,
     Param,
+    field,
     Task,
+    PathGenerator,
     Config,
-    pathoption,
     STDOUT,
     cache,
 )
@@ -17,9 +19,16 @@ class SimpleTask(Task):
     def execute(self):
         print(self.x)  # noqa: T201
+    def task_outputs(self, dep):
+        return dep(SimpleTaskOutput.C(task=self))
+class SimpleTaskOutput(Config):
+    task: Param[SimpleTask]
-@pathoption("out", STDOUT)
 class Say(Task):
+    out: Meta[Path] = field(default_factory=PathGenerator(STDOUT))
     word: Param[str]
     def execute(self):
@@ -38,19 +47,20 @@ class Concat(Task):
         print(" ".join(says))  # noqa: T201
-@param("x", type=int)
 class ForeignClassB1(Config):
-    pass
+    x: Param[int]
-@param("b", type=ForeignClassB1)
 class ForeignTaskA(Task):
+    b: Param[ForeignClassB1]
     def execute(self):
         print(self.b.x)  # noqa: T201
-@pathoption("wait", "wait")
 class Fail(Task):
+    wait: Meta[Path] = field(default_factory=PathGenerator("wait"))
     def execute(self):
         while not self.wait.is_file():
             time.sleep(0.01)
@@ -64,14 +74,16 @@ class Fail(Task):
             out.write("hello")
-@param("fail", Fail)
 class FailConsumer(Task):
+    fail: Param[Fail]
     def execute(self):
         return True
-@param("a", int)
 class Method(Task):
+    a: Param[int]
     def execute(self):
         assert self.a == 1
@@ -92,7 +104,8 @@ class CacheConfig(Config):
         return path.read_text()
-@param("data", type=CacheConfig)
 class CacheConfigTask(Task):
+    data: Param[CacheConfig]
     def execute(self):
         assert self.data.get() == "hello"

experimaestro/tests/tasks/foreign.py CHANGED Viewed

@@ -1,8 +1,6 @@
-from experimaestro import param, config
+from experimaestro import Param
 from .all import ForeignClassB1
-@param("y", type=int)
-@config()
 class ForeignClassB2(ForeignClassB1):
-    pass
+    y: Param[int]

experimaestro/tests/test_dependencies.py CHANGED Viewed

@@ -85,9 +85,3 @@ def test_dependencies_inner_task_output(xp):
     a = task_a.submit()
     b = Inner_TaskB(param_a=a).submit()
     check_dependencies(b, task_a)
-def test_dependencies_pre_task(xp):
-    a = TaskA().submit()
-    a2 = TaskA().add_pretasks(a).submit()
-    check_dependencies(a2, a)

experimaestro 1.6.1__py3-none-any.whl → 1.15.2__py3-none-any.whl

experimaestro 1.6.1py3-none-any.whl → 1.15.2py3-none-any.whl