diff --git a/aixplain/factories/model_factory.py b/aixplain/factories/model_factory.py index 0aa9d981..cf75dd51 100644 --- a/aixplain/factories/model_factory.py +++ b/aixplain/factories/model_factory.py @@ -69,6 +69,7 @@ def _create_model_from_response(cls, response: Dict) -> Model: function=Function(response["function"]["id"]), parameters=parameters, is_subscribed=True if "subscription" in response else False, + version=response["version"]["id"], ) @classmethod diff --git a/aixplain/modules/model.py b/aixplain/modules/model.py index 69109565..a4264107 100644 --- a/aixplain/modules/model.py +++ b/aixplain/modules/model.py @@ -59,7 +59,7 @@ def __init__( description: Text = "", api_key: Optional[Text] = None, supplier: Text = "aiXplain", - version: Text = "1.0", + version: Optional[Text] = None, function: Optional[Text] = None, is_subscribed: bool = False, **additional_info,