Fixed test, scales must be float

Author: NikolasMarkou

tests/test_optimizers.py

@@ -1178,20 +1178,25 @@ def test_cast_back_to_back_non_const_mixed_types(self):
                              "Cast", 5)
 
     def test_upsample_all_ones_removed(self):
-        node1 = helper.make_node("Upsample", ["X"], ["Y"], scales=[1, 1, 1, 1], name="upsample1")
+        node1 = helper.make_node(
+            "Upsample",
+            ["X"],
+            ["Y"],
+            scales=[1., 1., 1., 1.],
+            name="upsample1")
 
         graph = helper.make_graph(
             [node1],
             "test_upsample_all_ones",
-            [helper.make_tensor_value_info("X", TensorProto.FLOAT, (32, 16))],
-            [helper.make_tensor_value_info("Y", TensorProto.FLOAT, (32, 16))],
+            [helper.make_tensor_value_info("X", TensorProto.FLOAT, (1, 32, 32, 1))],
+            [helper.make_tensor_value_info("Y", TensorProto.FLOAT, (1, 32, 32, 1))],
         )
 
         model_proto = self.make_model(graph, producer_name="onnx-tests")
 
         self.run_and_compare(
             ["Y"],
-            {"X": np.random.randn(32, 16).astype(np.float32)},
+            {"X": np.random.randn(1, 32, 32, 1).astype(np.float32)},
             model_proto,
             "Upsample",
             0)