Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 4 additions & 5 deletions tests/unit/aiplatform/test_logdir_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
10 changes: 4 additions & 6 deletions tests/unit/aiplatform/test_uploader_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down
Loading