From 7ab25421925e9b9c23ed57767ac81ee6ae52b4c8 Mon Sep 17 00:00:00 2001 From: Matej Aleksandrov Date: Mon, 12 Jan 2026 05:48:37 -0800 Subject: [PATCH] fix: Enable tests for Python 3.13 PiperOrigin-RevId: 855191213 --- tests/unit/aiplatform/test_logdir_loader.py | 9 ++++----- tests/unit/aiplatform/test_uploader_utils.py | 10 ++++------ 2 files changed, 8 insertions(+), 11 deletions(-) diff --git a/tests/unit/aiplatform/test_logdir_loader.py b/tests/unit/aiplatform/test_logdir_loader.py index c2e22a2b19..44a66a79d8 100644 --- a/tests/unit/aiplatform/test_logdir_loader.py +++ b/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 diff --git a/tests/unit/aiplatform/test_uploader_utils.py b/tests/unit/aiplatform/test_uploader_utils.py index 6b11986aa8..5b4356ebc0 100644 --- a/tests/unit/aiplatform/test_uploader_utils.py +++ b/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: