Skip to content

Commit

Permalink
fix(kedro.session): use only root package name for determining version
Browse files Browse the repository at this point in the history
  • Loading branch information
lariel-fernandes committed Oct 21, 2024
1 parent 022f0d2 commit 796bc23
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion src/mlopus/kedro/session.py
Original file line number Diff line number Diff line change
Expand Up @@ -103,7 +103,10 @@ def __init__(
if self._package_name is None: # resolve package name from project metadata if not specified
self._package_name = toml.load(self._project_path / "pyproject.toml")["tool"]["kedro"]["package_name"]

self._store["pkg"] = {"name": self._package_name, "version": packaging.get_dist(self._package_name).version}
self._store["pkg"] = {
"name": self._package_name,
"version": packaging.get_dist(self._package_name.split(".")[0]).version,
}

self._store["uuid"] = self.uuid = str(uuid.uuid4()) # generate UUID (not datetime-bound like the session ID)

Expand Down

0 comments on commit 796bc23

Please sign in to comment.