fix: Enable tests for Python 3.13

PiperOrigin-RevId: 855191213

Author: AleksMat

tests/unit/aiplatform/test_logdir_loader.py

@@ -21,15 +21,14 @@
 import shutil
 import tempfile
 
-import sys
-
 import pytest
 
-if sys.version_info >= (3, 13):
-    pytest.skip("Tensorflow not available for Python 3.13+", allow_module_level=True)
+try:
+    import tensorflow as tf
+except ImportError:
+    pytest.skip("Tensorflow not available", allow_module_level=True)
 
 from google.cloud.aiplatform.tensorboard import logdir_loader
-import tensorflow as tf
 
 from tensorboard.backend.event_processing import directory_loader
 from tensorboard.backend.event_processing import event_file_loader

tests/unit/aiplatform/test_uploader_utils.py

@@ -20,14 +20,12 @@
 from unittest import mock
 
 from absl.testing import absltest
-import sys
-
 import pytest
 
-if sys.version_info >= (3, 13):
-    pytest.skip("Tensorboard not available for Python 3.13+", allow_module_level=True)
-
-from google.cloud.aiplatform.tensorboard import uploader_utils
+try:
+    from google.cloud.aiplatform.tensorboard import uploader_utils
+except ImportError:
+    pytest.skip("Tensorboard not available", allow_module_level=True)
 
 
 class FakeTime: