docs

Author: dreaming-panda

docs/conf.py

@@ -35,7 +35,8 @@
 autodoc_mock_imports = [
     "torch",
     "triton",
-    "numpy"
+    "numpy",
+    "vortex_torch_C"
 ]
 
 

vortex_torch/abs/tensor.py

@@ -1,3 +1,4 @@
+from __future__ import annotations
 import torch
 from enum import Enum
 from typing import Any, Iterable

vortex_torch/flow/loader.py

@@ -1,4 +1,4 @@
-# loader.py
+from __future__ import annotations
 import os
 import uuid
 import types

vortex_torch/indexer/output_func.py

@@ -1,5 +1,5 @@
 import torch
-from typing import Any, Tuple, Dict, Callable, Optional
+from typing import Dict, Callable, Optional
 from ..abs import vOp
 from vortex_torch_C import topk_output
 from .context import Context

vortex_torch/indexer/utils_sglang.py

@@ -8,7 +8,7 @@
 Chunkwise_HN2NH_Transpose,
 Chunkwise_HN2NH_Transpose_FA3
 )
-from typing import Tuple, Any
+from typing import Tuple
 from .context import Context