From 13fd7bd72149826cdb78a7f6921ea6558c1d70ab Mon Sep 17 00:00:00 2001
From: dc-dc-dc <dgcruz983@gmail.com>
Date: Fri, 22 Dec 2023 01:25:42 -0500
Subject: [PATCH] small fixes

---
 python/src/load.cpp | 2 ++
 python/src/load.h   | 5 ++++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/python/src/load.cpp b/python/src/load.cpp
index 492a40b41..18157dfc0 100644
--- a/python/src/load.cpp
+++ b/python/src/load.cpp
@@ -232,6 +232,8 @@ array mlx_load_npy_helper(py::object file, StreamOrDevice s) {
     }
     return {arr};
   }
+  throw std::invalid_argument(
+      "[load_npy] Input must be a file-like object, string, or pathlib.Path");
 }
 
 DictOrArray mlx_load_helper(
diff --git a/python/src/load.h b/python/src/load.h
index 9103678bf..e34977151 100644
--- a/python/src/load.h
+++ b/python/src/load.h
@@ -22,7 +22,10 @@ void mlx_save_safetensor_helper(
     py::dict d,
     bool retain_graph = true);
 
-DictOrArray mlx_load_helper(py::object file, StreamOrDevice s);
+DictOrArray mlx_load_helper(
+    py::object file,
+    std::optional<std::string> format,
+    StreamOrDevice s);
 void mlx_save_helper(
     py::object file,
     array a,