Skip to content

Commit

Permalink
Fixes up the sdk so that we can use local modules for hamilton and it…
Browse files Browse the repository at this point in the history
… won't break
  • Loading branch information
elijahbenizzy committed Jun 10, 2024
1 parent 1b875e1 commit 3bb79ab
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions ui/sdk/src/hamilton_sdk/driver.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,8 +82,10 @@ def safe_getmembers(module):
)
continue
# Check if the module is in the same top level package
if value.__package__ != module.__package__ and not value.__package__.startswith(
module.__package__
if (
value.__package__ != module.__package__
and module.__package__ is not None
and value.__package__.startswith(module.__package__)
):
logger.debug(
f"Skipping hash for module {value.__name__} because it is in a different "
Expand Down

0 comments on commit 3bb79ab

Please sign in to comment.