fix: raise exception when type of DocList is object

punndcoder28 · punndcoder28 · commit 862181783471 · 2023-09-18T18:14:30.000+05:30
Signed-off-by: punndcoder28 &lt;puneethk.2899@gmail.com&gt;
diff --git a/docarray/array/doc_list/doc_list.py b/docarray/array/doc_list/doc_list.py
@@ -337,8 +337,14 @@ def __class_getitem__(cls, item: Union[Type[BaseDoc], TypeVar, str]):
 
         if isinstance(item, type) and safe_issubclass(item, BaseDoc):
             return AnyDocArray.__class_getitem__.__func__(cls, item)  # type: ignore
-        else:
-            return super().__class_getitem__(item)
+        if (
+            isinstance(item, object)
+            and not isinstance(item, TypeVar)
+            and item is not Any
+        ):
+            raise TypeError('Expecting a type, got object instead')
+
+        return super().__class_getitem__(item)
 
     def __repr__(self):
         return AnyDocArray.__repr__(self)  # type: ignore
diff --git a/tests/units/array/test_array.py b/tests/units/array/test_array.py
@@ -487,3 +487,13 @@ def test_legacy_doc():
     newDoc = LegacyDocument()
     da = DocList[LegacyDocument]([newDoc])
     da.summary()
+
+
+def test_parameterize_list():
+    from docarray import DocList, BaseDoc
+
+    with pytest.raises(TypeError) as excinfo:
+        doc = DocList[BaseDoc()]
+
+    assert str(excinfo.value) == 'Expecting a type, got object instead'
+    assert doc is None
