fixes and cleanup

Jay Bhambhani · Jay Bhambhani · commit ed98b77cfce3 · 2025-02-28T19:33:36.000-05:00
diff --git a/docarray/index/backends/opensearchv2.py b/docarray/index/backends/opensearchv2.py
@@ -91,7 +91,6 @@ def __init__(self, db_config=None, **kwargs):
 
         self._logger.debug('Mappings have been updated with db_config.index_mappings')
 
-        print('INFO', self._column_infos)
         for col_name, col in self._column_infos.items():
 
             if safe_issubclass(col.docarray_type, AnyDocArray):
@@ -342,7 +341,6 @@ def python_type_to_db_type(self, python_type: Type) -> Any:
             or None if ``python_type`` is not supported.
         """
         self._logger.debug(f'Mapping Python type {python_type} to database type')
-        print('PYTHON_TYPE', python_type)
 
         for allowed_type in OPENSEARCH_PY_VEC_TYPES:
             if safe_issubclass(python_type, allowed_type):
@@ -477,7 +475,6 @@ def execute_query(self, query: Dict[str, Any], *args, **kwargs) -> Any:
             self._logger.error(err_msg)
             raise ValueError(err_msg)
 
-        print('EXECUTING', query)
         resp = self._client.search(index=self.index_name, body=query)
         docs, scores = self._format_response(resp)
 
diff --git a/tests/index/opensearch/test_subindex_os.py b/tests/index/opensearch/test_subindex_os.py
@@ -37,7 +37,7 @@ def index():
                 [
                     SimpleDoc(
                         id=f'docs-{i}-{j}',
-                        simple_tens=np.ones(10) * (j + 1),
+                        simple_tens=np.ones(10),
                         simple_text=f'hello {j}',
                     )
                     for j in range(5)
@@ -51,18 +51,18 @@ def index():
                             [
                                 SimpleDoc(
                                     id=f'list_docs-docs-{i}-{j}-{k}',
-                                    simple_tens=np.ones(10) * (k + 1),
+                                    simple_tens=np.ones(10),
                                     simple_text=f'hello {k}',
                                 )
                                 for k in range(5)
                             ]
                         ),
                         simple_doc=SimpleDoc(
                             id=f'list_docs-simple_doc-{i}-{j}',
-                            simple_tens=np.ones(10) * (j + 1),
+                            simple_tens=np.ones(10),
                             simple_text=f'hello {j}',
                         ),
-                        list_tens=np.ones(20) * (j + 1),
+                        list_tens=np.ones(20),
                     )
                     for j in range(5)
                 ]

Original file line number	Diff line number	Diff line change
`@@ -37,7 +37,7 @@ def index():`
`37`	`37`	`[`
`38`	`38`	`SimpleDoc(`
`39`	`39`	`id=f'docs-{i}-{j}',`
`40`		`- simple_tens=np.ones(10) * (j + 1),`
	`40`	`+ simple_tens=np.ones(10),`
`41`	`41`	`simple_text=f'hello {j}',`
`42`	`42`	`)`
`43`	`43`	`for j in range(5)`
`@@ -51,18 +51,18 @@ def index():`
`51`	`51`	`[`
`52`	`52`	`SimpleDoc(`
`53`	`53`	`id=f'list_docs-docs-{i}-{j}-{k}',`
`54`		`- simple_tens=np.ones(10) * (k + 1),`
	`54`	`+ simple_tens=np.ones(10),`
`55`	`55`	`simple_text=f'hello {k}',`
`56`	`56`	`)`
`57`	`57`	`for k in range(5)`
`58`	`58`	`]`
`59`	`59`	`),`
`60`	`60`	`simple_doc=SimpleDoc(`
`61`	`61`	`id=f'list_docs-simple_doc-{i}-{j}',`
`62`		`- simple_tens=np.ones(10) * (j + 1),`
	`62`	`+ simple_tens=np.ones(10),`
`63`	`63`	`simple_text=f'hello {j}',`
`64`	`64`	`),`
`65`		`- list_tens=np.ones(20) * (j + 1),`
	`65`	`+ list_tens=np.ones(20),`
`66`	`66`	`)`
`67`	`67`	`for j in range(5)`
`68`	`68`	`]`