Skip to content

Commit

Permalink
chore: skip tensorflow tests (#2570)
Browse files Browse the repository at this point in the history
  • Loading branch information
eddyxu committed Jul 6, 2024
1 parent 36c58f0 commit 1fdcdd9
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions python/python/tests/test_tf.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,12 +4,16 @@
import os
import warnings

import lance
import ml_dtypes
import numpy as np
import pandas as pd
import pyarrow as pa
import pytest
from lance.arrow import BFloat16Type, ImageArray, bfloat16_array
from lance.fragment import LanceFragment

pytest.skip("Skip tensorflow tests", allow_module_level=True)

try:
with warnings.catch_warnings():
Expand All @@ -22,15 +26,13 @@
allow_module_level=True,
)

import lance
from lance.fragment import LanceFragment
from lance.tf.data import (
from lance.tf.data import ( # noqa: E402
from_lance,
from_lance_batches,
lance_fragments,
lance_take_batches,
)
from lance.tf.tfrecord import infer_tfrecord_schema, read_tfrecord
from lance.tf.tfrecord import infer_tfrecord_schema, read_tfrecord # noqa: E402


@pytest.fixture
Expand Down

0 comments on commit 1fdcdd9

Please sign in to comment.