@SaraKingGH Two answers to this.

First case: The import problem

In the first instance, could you tell us which context you executed the script in? To make it work 'as is', you should run it within the examples/scikit directory. From scratch, do the following:

git clone https://github.com/maiot-io/zenml.git

Before continuing, either install the zenml pip package or install it from the cloned repo. Then:

cd zenml
zenml init
cd examples/scikit

python run.py

This is working for me personally, and looking at your screenshot also seems to be what you're doing. Not sure why you had to resolve from examples.scikit.step.trainer import MyScikitTrainer in run.py to from step.trainer import MyScikitTrainer. Could you try repeating the above steps and tell if the problems persists?

Second case: The Ref HEAD problem

The screenshot you shared seems to indicate that there is a file called fun.py within the directory. And that seems to be the cause of the problem: As it seems not to be tracked by Git at all, we are having a hard error when it tries to resolve the step. Read more about how Git resolution works here: https://docs.zenml.io/repository/integration-with-git.html .

This is not intended behavior, and seems to be a bug. Further investigation is warranted. For now, please try and commit fun.py before running and it should work better. Let me know if any of that helps!

from zenml.

@SaraKingGH Update: After further investigation, my above diagnosis seems flawed.You were indeed correct to point out that changing from from examples.scikit.step.trainer import MyScikitTrainer in run.py of the scikit example to from step.trainer import MyScikitTrainer would yield the error you showed. This was becuase of the logic of resolving relative imports in the git wrapper of zenml. We have no fixed this issue with PR #59. After merge and next release (hopefully today or tomorrow), please installed the latest zenml and see if it works now :-). Thank you!

from zenml.

@SaraKingGH Update: After further investigation, my above diagnosis seems flawed.You were indeed correct to point out that changing from from examples.scikit.step.trainer import MyScikitTrainer in run.py of the scikit example to from step.trainer import MyScikitTrainer would yield the error you showed. This was becuase of the logic of resolving relative imports in the git wrapper of zenml. We have no fixed this issue with PR #59. After merge and next release (hopefully today or tomorrow), please installed the latest zenml and see if it works now :-). Thank you!

sounds great Tahir, thanks for letting me know.:)

from zenml.