experimaestro 1.8.0rc6__tar.gz → 1.8.3__tar.gz

{experimaestro-1.8.0rc6 → experimaestro-1.8.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: experimaestro
-Version: 1.8.0rc6
+Version: 1.8.3
 Summary: "Experimaestro is a computer science experiment manager"
 License: GPL-3
 Keywords: experiment manager
@@ -28,7 +28,7 @@ Requires-Dist: docstring-parser (>=0.15,<0.16)
 Requires-Dist: fasteners (>=0.19,<0.20)
 Requires-Dist: flask (>=2.3,<3.0)
 Requires-Dist: flask-socketio (>=5.3,<6.0)
-Requires-Dist: gevent (>=23.9,<24.0)
+Requires-Dist: gevent (>=24.11.1,<25.0.0)
 Requires-Dist: gevent-websocket (>=0.10,<0.11)
 Requires-Dist: huggingface-hub (>0.17)
 Requires-Dist: humanfriendly (>=10,<11)
@@ -144,11 +144,11 @@ def cli(port, workdir, sleeptime):
     # Sets the working directory and the name of the xp
     with experiment(workdir, "helloworld", port=port) as xp:
         # Submit the tasks
-        hello = Say(word="hello", sleeptime=sleeptime).submit()
-        world = Say(word="world", sleeptime=sleeptime).submit()
+        hello = Say.C(word="hello", sleeptime=sleeptime).submit()
+        world = Say.C(word="world", sleeptime=sleeptime).submit()
 
         # Concat will depend on the two first tasks
-        Concat(strings=[hello, world], sleeptime=sleeptime).tag("y", 1).submit()
+        Concat.C(strings=[hello, world], sleeptime=sleeptime).tag("y", 1).submit()
 
 
 if __name__ == "__main__":

{experimaestro-1.8.0rc6 → experimaestro-1.8.3}/README.md

@@ -93,11 +93,11 @@ def cli(port, workdir, sleeptime):
     # Sets the working directory and the name of the xp
     with experiment(workdir, "helloworld", port=port) as xp:
         # Submit the tasks
-        hello = Say(word="hello", sleeptime=sleeptime).submit()
-        world = Say(word="world", sleeptime=sleeptime).submit()
+        hello = Say.C(word="hello", sleeptime=sleeptime).submit()
+        world = Say.C(word="world", sleeptime=sleeptime).submit()
 
         # Concat will depend on the two first tasks
-        Concat(strings=[hello, world], sleeptime=sleeptime).tag("y", 1).submit()
+        Concat.C(strings=[hello, world], sleeptime=sleeptime).tag("y", 1).submit()
 
 
 if __name__ == "__main__":

{experimaestro-1.8.0rc6 → experimaestro-1.8.3}/pyproject.toml

@@ -20,7 +20,7 @@ include = [
     "src/experimaestro/mkdocs/style.css",
     { path="src/experimaestro/server/data/*", format=['sdist', 'wheel']}
 ]
-version = "1.8.0-rc6"
+version = "1.8.3"
 repository = "https://github.com/experimaestro/experimaestro-python"
 documentation = "https://experimaestro-python.readthedocs.io/"
 
@@ -55,7 +55,7 @@ sortedcontainers = "^2.4"
 pyparsing = "^3.1"
 humanfriendly = "^10"
 huggingface-hub = ">0.17"
-gevent = "^23.9"
+gevent = "^24.11.1"
 gevent-websocket = "^0.10"
 flask = "^2.3"
 flask-socketio = "^5.3"
@@ -122,7 +122,7 @@ warn_unused_ignores = true
 
 [tool.commitizen]
 name = "cz_conventional_commits"
-version = "1.8.0rc6"
+version = "1.8.3"
 changelog_start_rev = "v1.0.0"
 tag_format = "v$major.$minor.$patch$prerelease"
 # update_changelog_on_bump = true

experimaestro-1.8.3/src/experimaestro/core/identifier.py

@@ -0,0 +1,296 @@
+from contextlib import contextmanager
+from enum import Enum
+from functools import cached_property
+import hashlib
+import logging
+import os
+import struct
+from typing import Optional
+from experimaestro.core.objects import Config
+
+
+class ConfigPath:
+    """Used to keep track of cycles when computing a hash"""
+
+    def __init__(self):
+        self.loops: list[bool] = []
+        """Indicates whether a loop was detected up to this node"""
+
+        self.config2index = {}
+        """Associate an index in the list with a configuration"""
+
+    def detect_loop(self, config) -> Optional[int]:
+        """If there is a loop, return the relative index and update the path"""
+        index = self.config2index.get(id(config), None)
+        if index is not None:
+            for i in range(index, self.depth):
+                self.loops[i] = True
+            return self.depth - index
+        return None
+
+    def has_loop(self):
+        return self.loops[-1]
+
+    @property
+    def depth(self):
+        return len(self.loops)
+
+    @contextmanager
+    def push(self, config):
+        config_id = id(config)
+        assert config_id not in self.config2index
+
+        self.config2index[config_id] = self.depth
+        self.loops.append(False)
+
+        try:
+            yield
+        finally:
+            self.loops.pop()
+            del self.config2index[config_id]
+
+
+hash_logger = logging.getLogger("xpm.hash")
+
+
+def is_ignored(value):
+    """Returns True if the value should be ignored by itself"""
+    return value is not None and isinstance(value, Config) and (value.__xpm__.meta)
+
+
+def remove_meta(value):
+    """Cleanup a dict/list by removing ignored values"""
+    if isinstance(value, list):
+        return [el for el in value if not is_ignored(el)]
+    if isinstance(value, dict):
+        return {key: value for key, value in value.items() if not is_ignored(value)}
+    return value
+
+
+class Identifier:
+    def __init__(self, main: bytes):
+        self.main = main
+        self.has_loops = False
+
+    @cached_property
+    def all(self):
+        """Returns the overall identifier"""
+        return self.main
+
+    def __hash__(self) -> int:
+        return hash(self.main)
+
+    def state_dict(self):
+        return self.main.hex()
+
+    def __eq__(self, other: object):
+        if not isinstance(other, Identifier):
+            return False
+        return self.main == other.main
+
+    @staticmethod
+    def from_state_dict(data: dict[str, str] | str):
+        if isinstance(data, str):
+            return Identifier(bytes.fromhex(data))
+
+        return Identifier(bytes.fromhex(data["main"]))
+
+    def __repr__(self):
+        return self.main.hex()
+
+
+class IdentifierComputer:
+    """This class is in charge of computing a config/task identifier"""
+
+    OBJECT_ID = b"\x00"
+    INT_ID = b"\x01"
+    FLOAT_ID = b"\x02"
+    STR_ID = b"\x03"
+    PATH_ID = b"\x04"
+    NAME_ID = b"\x05"
+    NONE_ID = b"\x06"
+    LIST_ID = b"\x07"
+    TASK_ID = b"\x08"
+    DICT_ID = b"\x09"
+    ENUM_ID = b"\x0a"
+    CYCLE_REFERENCE = b"\x0b"
+    INIT_TASKS = b"\x0c"
+
+    def __init__(self, config: "Config", config_path: ConfigPath, *, version=None):
+        # Hasher for parameters
+        self._hasher = hashlib.sha256()
+        self.config = config
+        self.config_path = config_path
+        self.version = version or int(os.environ.get("XPM_HASH_COMPUTER", 2))
+        if hash_logger.isEnabledFor(logging.DEBUG):
+            hash_logger.debug(
+                "starting hash (%s): %s", hash(str(self.config)), self.config
+            )
+
+    def identifier(self) -> Identifier:
+        main = self._hasher.digest()
+        if hash_logger.isEnabledFor(logging.DEBUG):
+            hash_logger.debug("hash (%s): %s", hash(str(self.config)), str(main))
+        return Identifier(main)
+
+    def _hashupdate(self, bytes: bytes):
+        """Update the hash computers with some bytes"""
+        if hash_logger.isEnabledFor(logging.DEBUG):
+            hash_logger.debug(
+                "updating hash (%s): %s", hash(str(self.config)), str(bytes)
+            )
+        self._hasher.update(bytes)
+
+    def update(self, value, *, myself=False):  # noqa: C901
+        """Update the hash
+
+        :param value: The value to add to the hash
+        :param myself: True if the value is the configuration for which we wish
+            to compute the identifier, defaults to False
+        :raises NotImplementedError: If the value cannot be processed
+        """
+        if value is None:
+            self._hashupdate(IdentifierComputer.NONE_ID)
+        elif isinstance(value, float):
+            self._hashupdate(IdentifierComputer.FLOAT_ID)
+            self._hashupdate(struct.pack("!d", value))
+        elif isinstance(value, int):
+            self._hashupdate(IdentifierComputer.INT_ID)
+            self._hashupdate(struct.pack("!q", value))
+        elif isinstance(value, str):
+            self._hashupdate(IdentifierComputer.STR_ID)
+            self._hashupdate(value.encode("utf-8"))
+        elif isinstance(value, list):
+            values = [el for el in value if not is_ignored(el)]
+            self._hashupdate(IdentifierComputer.LIST_ID)
+            self._hashupdate(struct.pack("!d", len(values)))
+            for x in values:
+                self.update(x)
+        elif isinstance(value, Enum):
+            self._hashupdate(IdentifierComputer.ENUM_ID)
+            k = value.__class__
+            self._hashupdate(
+                f"{k.__module__}.{k.__qualname__ }:{value.name}".encode("utf-8"),
+            )
+        elif isinstance(value, dict):
+            self._hashupdate(IdentifierComputer.DICT_ID)
+            items = [
+                (key, value) for key, value in value.items() if not is_ignored(value)
+            ]
+            items.sort(key=lambda x: x[0])
+            for key, value in items:
+                self.update(key)
+                self.update(value)
+
+        # Handles configurations
+        elif isinstance(value, Config):
+            # Encodes the identifier
+            self._hashupdate(IdentifierComputer.OBJECT_ID)
+
+            # If we encode another config, then
+            if not myself:
+                if loop_ix := self.config_path.detect_loop(value):
+                    # Loop detected: use cycle reference
+                    self._hashupdate(IdentifierComputer.CYCLE_REFERENCE)
+                    self._hashupdate(struct.pack("!q", loop_ix))
+
+                else:
+                    # Just use the object identifier
+                    value_id = IdentifierComputer.compute(
+                        value, version=self.version, config_path=self.config_path
+                    )
+                    self._hashupdate(value_id.all)
+
+                # And that's it!
+                return
+
+            # Process tasks
+            if value.__xpm__.task is not None and (value.__xpm__.task is not value):
+                hash_logger.debug("Computing hash for task %s", value.__xpm__.task)
+                self._hashupdate(IdentifierComputer.TASK_ID)
+                self.update(value.__xpm__.task)
+
+            xpmtype = value.__xpmtype__
+            self._hashupdate(xpmtype.identifier.name.encode("utf-8"))
+
+            # Process arguments (sort by name to ensure uniqueness)
+            arguments = sorted(xpmtype.arguments.values(), key=lambda a: a.name)
+            for argument in arguments:
+                # Ignored argument
+                if argument.ignored:
+                    argvalue = value.__xpm__.values.get(argument.name, None)
+
+                    # ... unless meta is set to false
+                    if (
+                        argvalue is None
+                        or not isinstance(argvalue, Config)
+                        or (argvalue.__xpm__.meta is not False)
+                    ):
+                        continue
+
+                if argument.generator:
+                    continue
+
+                # Argument value
+                # Skip if the argument is not a constant, and
+                # - optional argument: both value and default are None
+                # - the argument value is equal to the default value
+                argvalue = getattr(value, argument.name, None)
+                if not argument.constant and (
+                    (
+                        not argument.required
+                        and argument.default is None
+                        and argvalue is None
+                    )
+                    or (
+                        argument.default is not None
+                        and argument.default == remove_meta(argvalue)
+                    )
+                ):
+                    # No update if same value (and not constant)
+                    continue
+
+                if (
+                    argvalue is not None
+                    and isinstance(argvalue, Config)
+                    and argvalue.__xpm__.meta
+                ):
+                    continue
+
+                # Hash name
+                self.update(argument.name)
+
+                # Hash value
+                self._hashupdate(IdentifierComputer.NAME_ID)
+                self.update(argvalue)
+
+        else:
+            raise NotImplementedError("Cannot compute hash of type %s" % type(value))
+
+    @staticmethod
+    def compute(
+        config: "Config", config_path: ConfigPath | None = None, version=None
+    ) -> Identifier:
+        """Compute the identifier for a configuration
+
+        :param config: the configuration for which we compute the identifier
+        :param config_path: used to track down cycles between configurations
+        :param version: version for the hash computation (None for the last one)
+        """
+
+        # Try to use the cached value first
+        # (if there are no loops)
+        if config.__xpm__._sealed:
+            identifier = config.__xpm__._raw_identifier
+            if identifier is not None and not identifier.has_loops:
+                return identifier
+
+        config_path = config_path or ConfigPath()
+
+        with config_path.push(config):
+            self = IdentifierComputer(config, config_path, version=version)
+            self.update(config, myself=True)
+            identifier = self.identifier()
+            identifier.has_loops = config_path.has_loop()
+
+        return identifier

experimaestro-1.8.3/src/experimaestro/core/objects/__init__.py

@@ -0,0 +1,44 @@
+from .config_walk import ConfigWalkContext, ConfigWalk
+from .config import (
+    ConfigMixin,
+    Config,
+    ConfigInformation,
+    Task,
+    LightweightTask,
+    WatchedOutput,
+    DependentMarker,
+    copyconfig,
+    setmeta,
+    cache,
+    logger,
+)
+
+from .config_utils import (
+    getqualattr,
+    add_to_path,
+    ObjectStore,
+    SealedError,
+    TaggedValue,
+)
+
+
+__all__ = [
+    "ConfigMixin",
+    "Config",
+    "ConfigInformation",
+    "ConfigWalkContext",
+    "ConfigWalk",
+    "Task",
+    "LightweightTask",
+    "ObjectStore",
+    "WatchedOutput",
+    "SealedError",
+    "DependentMarker",
+    "TaggedValue",
+    "getqualattr",
+    "copyconfig",
+    "setmeta",
+    "cache",
+    "add_to_path",
+    "logger",
+]