-
Notifications
You must be signed in to change notification settings - Fork 293
[Hub] Support adding hub steps to the serving graph #9133
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
davesh0812
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks very good
| @pytest.mark.skip( | ||
| reason="Remove this marker after steps are added to the functions repo" | ||
| ) | ||
| def test_serving_with_hub_steps(self): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
please lets have the same test also as sys test
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Well since it's uses an external resource (hub step) I think that it's enough to use just integ test.
Eyal-Danieli
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Nice!!
few comments
Eyal-Danieli
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Nice, please add __init__.py file under the new hub system test dir
| class Echo(BaseClass): | ||
| def __init__(self, name=None): | ||
| self.name = name |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
just a recommendation
class Echo(BaseClass):
def __init__(self, context, name=None):
super().__init__(context=context, name=name)
Eyal-Danieli
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
just resolve conflicts :)
# Conflicts: # mlrun/runtimes/nuclio/serving.py
📝 Description
Add support to adding hub steps to the serving graph by implement infra
HubTaskStepand modifying the methods for adding steps to the graph🛠️ Changes Made
HubTaskStepkind, which inherits fromTaskStepand implements aninit_objectwhich initialize the actual step from the dynamically imported hub step code.params_to_step()by adding another branch to createHubTaskStepin case theclass_nameparameter is a hub URL.ServingRuntime'sdeploy()andto_job()methods by adding a step of adding the requirements of the local steps with the requirements of the function✅ Checklist
🧪 Testing
New integration test was added to cover the changes in this PR.
🔗 References
🚨 Breaking Changes?
🔍️ Additional Notes