Fix the usage of tempfile in test_generate_data

Replace mktemp with mkstemp
microsoft · Feb 13, 2024 · f9a075f · f9a075f
1 parent 6954849
commit f9a075f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/data_generation/test_generate_data.py b/test/data_generation/test_generate_data.py
@@ -177,7 +177,7 @@ def test_generate_sem_variable_groups(five_variable_two_dim_metadata: VariablesM
 
 
 def test_generate_sem_numpy(five_node_graph: torch.Tensor, five_variable_one_dim_metadata: VariablesMetadata):
-    tmpfile = tempfile.mktemp() + ".npy"
+    tmpfile = tempfile.mkstemp(suffix=".npy")[1]
 
     np.save(tmpfile, five_node_graph.numpy())
     assert os.path.exists(tmpfile)