Small bug fix for serve integration test (#5363)

* Fixing serve unit tests

* Bug fix for inference

---------

Co-authored-by: rsareddy0329 <rsareddy0329@gmail.com>

Author: aviruthen

sagemaker-serve/src/sagemaker/serve/model_builder.py

@@ -943,23 +943,23 @@ def _is_model_customization(self) -> bool:
         # ModelTrainer with model customization
         if isinstance(self.model, ModelTrainer) and hasattr(self.model, '_latest_training_job'):
             # Check model_package_config first (new location)
-            if (hasattr(self.model._latest_training_job, 'model_package_config') and self.model._latest_training_job.model_package_config != Unassigned
-                    and getattr(self.model._latest_training_job.model_package_config, 'source_model_package_arn', Unassigned) != Unassigned):
+            if (hasattr(self.model._latest_training_job, 'model_package_config') and self.model._latest_training_job.model_package_config != Unassigned()
+                    and getattr(self.model._latest_training_job.model_package_config, 'source_model_package_arn', Unassigned()) != Unassigned()):
                 return True
             # Fallback to serverless_job_config (legacy location)
-            if (hasattr(self.model._latest_training_job, 'serverless_job_config') and self.model._latest_training_job.serverless_job_config != Unassigned
-                    and hasattr(self.model._latest_training_job, 'output_model_package_arn') and self.model._latest_training_job.output_model_package_arn!= Unassigned):
+            if (hasattr(self.model._latest_training_job, 'serverless_job_config') and self.model._latest_training_job.serverless_job_config != Unassigned()
+                    and hasattr(self.model._latest_training_job, 'output_model_package_arn') and self.model._latest_training_job.output_model_package_arn!= Unassigned()):
                 return True
 
         # BaseTrainer with model customization
         if isinstance(self.model, BaseTrainer) and hasattr(self.model, '_latest_training_job'):
             # Check model_package_config first (new location)
-            if (hasattr(self.model._latest_training_job, 'model_package_config') and self.model._latest_training_job.model_package_config != Unassigned
-                    and getattr(self.model._latest_training_job.model_package_config, 'source_model_package_arn', Unassigned) != Unassigned):
+            if (hasattr(self.model._latest_training_job, 'model_package_config') and self.model._latest_training_job.model_package_config != Unassigned()
+                    and getattr(self.model._latest_training_job.model_package_config, 'source_model_package_arn', Unassigned()) != Unassigned()):
                 return True
             # Fallback to serverless_job_config (legacy location)
-            if (hasattr(self.model._latest_training_job, 'serverless_job_config') and self.model._latest_training_job.serverless_job_config != Unassigned
-                    and hasattr(self.model._latest_training_job, 'output_model_package_arn') and self.model._latest_training_job.output_model_package_arn!= Unassigned):
+            if (hasattr(self.model._latest_training_job, 'serverless_job_config') and self.model._latest_training_job.serverless_job_config != Unassigned()
+                    and hasattr(self.model._latest_training_job, 'output_model_package_arn') and self.model._latest_training_job.output_model_package_arn!= Unassigned()):
                 return True
 
         return False