Skip to content

Commit

Permalink
fix(package): clean up the tag for extra packages
Browse files Browse the repository at this point in the history
  • Loading branch information
mostaphaRoudsari committed May 17, 2023
1 parent 9f7fc36 commit 21a43dd
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions pollination_dsl/package.py
Original file line number Diff line number Diff line change
Expand Up @@ -179,6 +179,13 @@ def load(package_name: str, baked: bool = False) -> Union[Plugin, BakedRecipe, R
)
else:
dep.tag = tag

# tag might return additional information for extra dependencies
# here is an example: "0.1.11 ; extra 'viz'"
# we try to clean up the tag from the extra tag
if dep.tag:
dep.tag = dep.tag.split('; extra')[0].strip()

else:
raise ValueError(
f'Error loading {package_name}. Package must be a DSL plugin or a DSL '
Expand Down

0 comments on commit 21a43dd

Please sign in to comment.