diff --git a/tests/logprob/test_transforms.py b/tests/logprob/test_transforms.py index 2ab30235b..1f9fc838a 100644 --- a/tests/logprob/test_transforms.py +++ b/tests/logprob/test_transforms.py @@ -249,7 +249,7 @@ class TestLocScaleRVTransform: @pytest.mark.parametrize( "rv_size, loc_type, addition", [ - (None, pt.scalar, True), + ((), pt.scalar, True), (2, pt.vector, False), ((2, 1), pt.col, True), ], diff --git a/tests/test_pytensorf.py b/tests/test_pytensorf.py index 1ce421455..f7efd5f6d 100644 --- a/tests/test_pytensorf.py +++ b/tests/test_pytensorf.py @@ -60,7 +60,7 @@ ) def test_pd_dataframe_as_tensor_variable(np_array: np.ndarray) -> None: df = pd.DataFrame(np_array) - np.testing.assert_array_equal(x=pt.as_tensor_variable(x=df).eval(), y=np_array) + np.testing.assert_array_equal(pt.as_tensor_variable(df).eval(), np_array) @pytest.mark.parametrize( @@ -69,7 +69,7 @@ def test_pd_dataframe_as_tensor_variable(np_array: np.ndarray) -> None: ) def test_pd_series_as_tensor_variable(np_array: np.ndarray) -> None: df = pd.Series(np_array) - np.testing.assert_array_equal(x=pt.as_tensor_variable(x=df).eval(), y=np_array) + np.testing.assert_array_equal(pt.as_tensor_variable(df).eval(), np_array) def test_pd_as_tensor_variable_multiindex() -> None: @@ -80,7 +80,7 @@ def test_pd_as_tensor_variable_multiindex() -> None: df = pd.DataFrame({"A": [12.0, 80.0, 30.0, 20.0], "B": [120.0, 700.0, 30.0, 20.0]}, index=index) np_array = np.array([[12.0, 80.0, 30.0, 20.0], [120.0, 700.0, 30.0, 20.0]]).T assert isinstance(df.index, pd.MultiIndex) - np.testing.assert_array_equal(x=pt.as_tensor_variable(x=df).eval(), y=np_array) + np.testing.assert_array_equal(pt.as_tensor_variable(df).eval(), np_array) class TestBroadcasting: