Update dataset_service_test.py

Fix E721
OpenMined · Dec 1, 2024 · 822110b · 822110b
1 parent ad939c6
commit 822110b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/packages/syft/tests/syft/service/dataset/dataset_service_test.py b/packages/syft/tests/syft/service/dataset/dataset_service_test.py
@@ -302,10 +302,10 @@ def test_upload_dataset_with_assets_of_different_data_types(
     res = root_datasite_client.upload_dataset(dataset)
     assert isinstance(res, SyftSuccess)
     assert len(root_datasite_client.api.services.dataset.get_all()) == 1
-    assert type(root_datasite_client.datasets[0].assets[0].data) == type(
+    assert type(root_datasite_client.datasets[0].assets[0].data) is type(
         different_data_types
     )
-    assert type(root_datasite_client.datasets[0].assets[0].mock) == type(
+    assert type(root_datasite_client.datasets[0].assets[0].mock) is type(
         different_data_types
     )