Fixed bug.

Author: lu-ohai

ads/model/deployment/model_deployment_infrastructure.py

@@ -223,7 +223,19 @@ def _load_default_properties(self) -> Dict:
         if NB_SESSION_OCID:
             try:
                 nb_session = DSCNotebookSession.from_ocid(NB_SESSION_OCID)
-                nb_config = nb_session.notebook_session_configuration_details
+            except Exception as e:
+                logger.warning(
+                    f"Error fetching details about Notebook "
+                    f"session: {NB_SESSION_OCID}. {e}"
+                )
+                logger.debug(traceback.format_exc())
+
+            nb_config = getattr(
+                nb_session,
+                "notebook_session_config_details",
+                getattr(nb_session, "notebook_session_configuration_details", None),
+            )
+            if nb_config:
                 defaults[self.CONST_SHAPE_NAME] = nb_config.shape
 
                 if nb_config.notebook_session_shape_config_details:
@@ -234,13 +246,6 @@ def _load_default_properties(self) -> Dict:
                         notebook_shape_config_details
                     )
 
-            except Exception as e:
-                logger.warning(
-                    f"Error fetching details about Notebook "
-                    f"session: {NB_SESSION_OCID}. {e}"
-                )
-                logger.debug(traceback.format_exc())
-
         return defaults
 
     @property

ads/model/generic_model.py

@@ -2137,7 +2137,7 @@ def deploy(
             or self.properties.deployment_ocpus
             or MODEL_DEPLOYMENT_INSTANCE_OCPUS,
             deployment_memory_in_gbs = existing_infrastructure.shape_config_details.get(
-                "memory_in_gbs", None
+                "memoryInGBs", None
             )
             or self.properties.deployment_memory_in_gbs
             or MODEL_DEPLOYMENT_INSTANCE_MEMORY_IN_GBS,