fix tests

Author: bencrabtree

src/sagemaker/jumpstart/estimator.py

@@ -616,6 +616,7 @@ def _validate_model_id_and_get_type_hook():
         self.tolerate_vulnerable_model = estimator_init_kwargs.tolerate_vulnerable_model
         self.instance_count = estimator_init_kwargs.instance_count
         self.region = estimator_init_kwargs.region
+        self.environment = estimator_init_kwargs.environment
         self.orig_predictor_cls = None
         self.role = estimator_init_kwargs.role
         self.sagemaker_session = estimator_init_kwargs.sagemaker_session

src/sagemaker/jumpstart/utils.py

@@ -1671,7 +1671,7 @@ def get_model_access_config(accept_eula: Optional[bool], environment: Optional[d
     Args:
         accept_eula (Optional[bool]): Whether or not the EULA was accepted, optionally passed in to Estimator.fit().
     """
-    env_var_eula = environment.get("accept_eula")
+    env_var_eula = environment.get("accept_eula") if environment else None
     if env_var_eula and accept_eula is not None:
         raise ValueError(
             "Cannot pass in both accept_eula and environment variables. "