Fix L0_implicit_state (#6427) (#6442)

Co-authored-by: Iman Tabrizian <iman.tabrizian@gmail.com>
triton-inference-server · Oct 17, 2023 · 2172606 · 2172606
1 parent 3d9cd82
commit 2172606
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/qa/common/gen_common.py b/qa/common/gen_common.py
@@ -24,6 +24,8 @@
 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
+from typing import List
+
 # Common utilities for model generation scripts
 import numpy as np
 
@@ -154,5 +156,5 @@ def np_to_torch_dtype(np_dtype):
     elif np_dtype == np.float64:
         return torch.double
     elif np_dtype == np_dtype_string:
-        return None  # Not supported in Torch
+        return List[str]
     return None