remove venv and reformat

wenyi-guo · wenyi-guo · commit b634f7888ab4 · 2025-11-26T02:38:47.000Z
diff --git a/.gitignore b/.gitignore
@@ -20,5 +20,3 @@ build/
 .idea/
 
 venv/
-venv_tf/
-venv_jax/
diff --git a/keras_rs/src/layers/embedding/embed_reduce_test.py b/keras_rs/src/layers/embedding/embed_reduce_test.py
@@ -8,7 +8,6 @@
 
 from keras_rs.src import testing
 from keras_rs.src.layers.embedding.embed_reduce import EmbedReduce
-from keras_rs.src.utils import tpu_test_utils
 
 try:
     import jax
diff --git a/keras_rs/src/layers/feature_interaction/dot_interaction_test.py b/keras_rs/src/layers/feature_interaction/dot_interaction_test.py
@@ -8,7 +8,6 @@
 from keras_rs.src.layers.feature_interaction.dot_interaction import (
     DotInteraction,
 )
-from keras_rs.src.utils import tpu_test_utils
 
 
 class DotInteractionTest(testing.TestCase, parameterized.TestCase):
@@ -88,7 +87,8 @@ def test_call(self, self_interaction, skip_gather, exp_output_idx):
             output,
             self.exp_outputs[exp_output_idx],
             tpu_atol=1e-2,
-            tpu_rtol=1e-2)
+            tpu_rtol=1e-2,
+        )
 
     def test_invalid_input_rank(self):
         rank_1_input = [ops.ones((3,)), ops.ones((3,))]
diff --git a/keras_rs/src/layers/feature_interaction/feature_cross_test.py b/keras_rs/src/layers/feature_interaction/feature_cross_test.py
@@ -6,7 +6,6 @@
 
 from keras_rs.src import testing
 from keras_rs.src.layers.feature_interaction.feature_cross import FeatureCross
-from keras_rs.src.utils import tpu_test_utils
 
 
 class FeatureCrossTest(testing.TestCase, parameterized.TestCase):
diff --git a/keras_rs/src/layers/retrieval/hard_negative_mining_test.py b/keras_rs/src/layers/retrieval/hard_negative_mining_test.py
@@ -6,7 +6,6 @@
 
 from keras_rs.src import testing
 from keras_rs.src.layers.retrieval import hard_negative_mining
-from keras_rs.src.utils import tpu_test_utils
 
 
 class HardNegativeMiningTest(testing.TestCase, parameterized.TestCase):
diff --git a/keras_rs/src/layers/retrieval/remove_accidental_hits_test.py b/keras_rs/src/layers/retrieval/remove_accidental_hits_test.py
@@ -6,7 +6,6 @@
 
 from keras_rs.src import testing
 from keras_rs.src.layers.retrieval import remove_accidental_hits
-from keras_rs.src.utils import tpu_test_utils
 
 
 class RemoveAccidentalHitsTest(testing.TestCase, parameterized.TestCase):
diff --git a/keras_rs/src/layers/retrieval/sampling_probability_correction_test.py b/keras_rs/src/layers/retrieval/sampling_probability_correction_test.py
@@ -6,7 +6,6 @@
 
 from keras_rs.src import testing
 from keras_rs.src.layers.retrieval import sampling_probability_correction
-from keras_rs.src.utils import tpu_test_utils
 
 
 class SamplingProbabilityCorrectionTest(
diff --git a/keras_rs/src/losses/list_mle_loss_test.py b/keras_rs/src/losses/list_mle_loss_test.py
@@ -6,7 +6,6 @@
 
 from keras_rs.src import testing
 from keras_rs.src.losses.list_mle_loss import ListMLELoss
-from keras_rs.src.utils import tpu_test_utils
 
 
 class ListMLELossTest(testing.TestCase, parameterized.TestCase):
diff --git a/keras_rs/src/losses/pairwise_hinge_loss_test.py b/keras_rs/src/losses/pairwise_hinge_loss_test.py
@@ -6,7 +6,6 @@
 
 from keras_rs.src import testing
 from keras_rs.src.losses.pairwise_hinge_loss import PairwiseHingeLoss
-from keras_rs.src.utils import tpu_test_utils
 
 
 class PairwiseHingeLossTest(testing.TestCase, parameterized.TestCase):
diff --git a/keras_rs/src/losses/pairwise_logistic_loss_test.py b/keras_rs/src/losses/pairwise_logistic_loss_test.py
@@ -6,7 +6,6 @@
 
 from keras_rs.src import testing
 from keras_rs.src.losses.pairwise_logistic_loss import PairwiseLogisticLoss
-from keras_rs.src.utils import tpu_test_utils
 
 
 class PairwiseLogisticLossTest(testing.TestCase, parameterized.TestCase):
diff --git a/keras_rs/src/losses/pairwise_mean_squared_error_test.py b/keras_rs/src/losses/pairwise_mean_squared_error_test.py
@@ -8,7 +8,6 @@
 from keras_rs.src.losses.pairwise_mean_squared_error import (
     PairwiseMeanSquaredError,
 )
-from keras_rs.src.utils import tpu_test_utils
 
 
 class PairwiseMeanSquaredErrorTest(testing.TestCase, parameterized.TestCase):
diff --git a/keras_rs/src/losses/pairwise_soft_zero_one_loss_test.py b/keras_rs/src/losses/pairwise_soft_zero_one_loss_test.py
@@ -8,7 +8,6 @@
 from keras_rs.src.losses.pairwise_soft_zero_one_loss import (
     PairwiseSoftZeroOneLoss,
 )
-from keras_rs.src.utils import tpu_test_utils
 
 
 class PairwiseSoftZeroOneLossTest(testing.TestCase, parameterized.TestCase):
diff --git a/keras_rs/src/metrics/dcg_test.py b/keras_rs/src/metrics/dcg_test.py
@@ -362,4 +362,4 @@ def test_model_evaluate(self):
         model.evaluate(
             x=keras.random.normal((2, 20)),
             y=keras.random.randint((2, 5), minval=0, maxval=4),
-        )
+        )
diff --git a/keras_rs/src/testing/test_case.py b/keras_rs/src/testing/test_case.py
@@ -5,11 +5,12 @@
 
 import keras
 import numpy as np
+import tensorflow as tf
 
 from keras_rs.src import types
-import tensorflow as tf
 from keras_rs.src.utils import tpu_test_utils
 
+
 class TestCase(unittest.TestCase):
     """TestCase class for all Keras Recommenders tests."""
 
