[fix] to 'long' for win

cocoshe · Nov 22, 2024 · 9637797 · 9637797
1 parent 201d7a1
commit 9637797
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/layer_tests/pytorch_tests/test_index_fill_.py b/tests/layer_tests/pytorch_tests/test_index_fill_.py
@@ -79,5 +79,5 @@ def test_index_fill_single_index(self, ie_device, precision, ir_version, input_d
         shape = self.input_tensor.shape
         max_idx = shape[dim]
         n_select = np.random.randint(1, max_idx + 1)
-        index = torch.from_numpy(np.random.choice(np.arange(0, max_idx), n_select, replace=False))
+        index = torch.from_numpy(np.random.choice(np.arange(0, max_idx), n_select, replace=False)).to(torch.long)
         self._test(*self.create_model(dim, index, values), ie_device, precision, ir_version)