From bd1a11bef7cac31cea9256b98e9cb6006dc708c7 Mon Sep 17 00:00:00 2001
From: dc-dc-dc <dgcruz983@gmail.com>
Date: Tue, 2 Jan 2024 18:14:29 -0500
Subject: [PATCH] removed IntOrIntVec

---
 python/src/ops.cpp | 3 ++-
 python/src/utils.h | 2 --
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/python/src/ops.cpp b/python/src/ops.cpp
index 9a76188fa..18151daad 100644
--- a/python/src/ops.cpp
+++ b/python/src/ops.cpp
@@ -3198,7 +3198,8 @@ void init_ops(py::module_& m) {
       "tensordot",
       [](const array& a,
          const array& b,
-         const IntOrIntVec& dims,
+         const std::variant<std::monostate, int, std::vector<std::vector<int>>>&
+             dims,
          StreamOrDevice s) {
         if (std::holds_alternative<std::monostate>(dims)) {
           return tensordot(a, b, 2, s);
diff --git a/python/src/utils.h b/python/src/utils.h
index 86da9f6db..5ac878979 100644
--- a/python/src/utils.h
+++ b/python/src/utils.h
@@ -15,8 +15,6 @@ namespace py = pybind11;
 using namespace mlx::core;
 
 using IntOrVec = std::variant<std::monostate, int, std::vector<int>>;
-using IntOrIntVec =
-    std::variant<std::monostate, int, std::vector<std::vector<int>>>;
 using ScalarOrArray = std::
     variant<py::bool_, py::int_, py::float_, std::complex<float>, py::object>;
 static constexpr std::monostate none{};