trainable variables fix

keras-team · Sep 8, 2022 · 02dda74 · 02dda74
1 parent e3bd717
commit 02dda74
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keras_cv/models/object_detection/retina_net/retina_net_test.py b/keras_cv/models/object_detection/retina_net/retina_net_test.py
@@ -210,9 +210,9 @@ def test_weights_contained_in_trainable_variables(self):
         _ = retina_net(xs)
         variable_names = [x.name for x in retina_net.trainable_variables]
         # classification_head
-        self.assertIn('RetinaNet/prediction_head/conv2d_8/kernel:0', variable_names)
+        self.assertIn("RetinaNet/prediction_head/conv2d_8/kernel:0", variable_names)
         # box_head
-        self.assertIn('RetinaNet/prediction_head_1/conv2d_11/kernel:0', variable_names)
+        self.assertIn("RetinaNet/prediction_head_1/conv2d_11/kernel:0", variable_names)
 
     def test_weights_change(self):
         bounding_box_format = "xywh"