chore: fix broken test

PiperOrigin-RevId: 858688708

Author: speedstorm1

tests/unit/aiplatform/test_metadata_models.py

@@ -18,6 +18,7 @@
 import datetime
 import os
 import pickle
+import sys
 from importlib import reload
 from unittest import mock
 from unittest.mock import patch
@@ -339,6 +340,7 @@ def setup_method(self):
         reload(initializer)
         reload(metadata)
         reload(aiplatform)
+        reload(xgb)
 
     def teardown_method(self):
         initializer.global_pool.shutdown(wait=True)