fix

dongmao zhang · dongmao zhang · commit dc7ced1842aa · 2025-01-08T07:21:58.000Z
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -24,6 +24,7 @@
 ]
 autodoc_mock_imports = [
     "infinistore._infinistore",
+    "torch",
 ]
 
 templates_path = ["_templates"]
diff --git a/infinistore/lib.py b/infinistore/lib.py
@@ -1,4 +1,4 @@
-from infinistore import _infinistore
+import infinistore._infinistore as _infinistore
 import torch
 import os
 from typing import List, Tuple

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-from infinistore import _infinistore`
	`1`	`+import infinistore._infinistore as _infinistore`
`2`	`2`	`import torch`
`3`	`3`	`import os`
`4`	`4`	`from typing import List, Tuple`