Revert to state of 'Fix missing range dtype'

Author: alungriffith

keras/src/backend/openvino/excluded_concrete_tests.txt

@@ -38,7 +38,6 @@ NumpyDtypeTest::test_logspace
 NumpyDtypeTest::test_matmul_
 NumpyDtypeTest::test_max
 NumpyDtypeTest::test_mean
-NumpyDtypeTest::test_median
 NumpyDtypeTest::test_meshgrid
 NumpyDtypeTest::test_minimum_python_types
 NumpyDtypeTest::test_multiply

keras/src/backend/openvino/numpy.py

@@ -1088,13 +1088,10 @@ def median(x, axis=None, keepdims=False):
             ov_opset.constant([0], Type.i32).output(0),
             ov_opset.constant([0], Type.i32).output(0),
         ).output(0)
-        x_rank_scalar = ov_opset.squeeze(
-            x_rank, ov_opset.constant([0], Type.i32).output(0)
-        ).output(0)
         axis_as_range = ov_opset.range(
-            ov_opset.constant(0, Type.i32).output(0),
-            x_rank_scalar,
-            ov_opset.constant(1, Type.i32).output(0),
+            ov_opset.constant([0], Type.i32).output(0),
+            x_rank,
+            ov_opset.constant([1], Type.i32).output(0),
             "i32",
         ).output(0)
         axis_compare = ov_opset.not_equal(
@@ -1145,13 +1142,10 @@ def median(x, axis=None, keepdims=False):
             ov_opset.constant([0], Type.i32).output(0),
             ov_opset.constant([0], Type.i32).output(0),
         ).output(0)
-        x_rank_scalar = ov_opset.squeeze(
-            x_rank, ov_opset.constant([0], Type.i32).output(0)
-        ).output(0)
         axis_as_range = ov_opset.range(
-            ov_opset.constant(0, Type.i32).output(0),
-            x_rank_scalar,
-            ov_opset.constant(1, Type.i32).output(0),
+            ov_opset.constant([0], Type.i32).output(0),
+            x_rank,
+            ov_opset.constant([1], Type.i32).output(0),
             "i32",
         ).output(0)
         ov_axis_positive = ov_opset.gather(