remove deprecated assertDictContainsSubset

PiperOrigin-RevId: 716573448
tensorflow · Jan 17, 2025 · 910ceed · 910ceed
1 parent 8f2c6a9
commit 910ceed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/research/object_detection/models/feature_map_generators_test.py b/research/object_detection/models/feature_map_generators_test.py
@@ -513,7 +513,7 @@ def graph_fn():
         'top_down/clip_by_value_6'
     ])
     op_names = {op.name: None for op in g.get_operations()}
-    self.assertDictContainsSubset(expected_added_operations, op_names)
+    self.assertEqual(op_names, {**op_names, **expected_added_operations})
 
   @unittest.skipIf(tf_version.is_tf2(), 'Skipping TF1.X only test.')
   def test_use_bounded_activations_clip_value(