Make pecos.xmr importable

amzn · Oct 17, 2024 · d3ec83b · d3ec83b
1 parent cfeb1d8
commit d3ec83b
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 0 deletions.
diff --git a/pecos/xmr/__init__.py b/pecos/xmr/__init__.py
@@ -0,0 +1,10 @@
+#  Copyright 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+#  with the License. A copy of the License is located at
+#
+#  http://aws.amazon.com/apache2.0/
+#
+#  or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES
+#  OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions
+#  and limitations under the License.
diff --git a/pecos/xmr/reranker/__init__.py b/pecos/xmr/reranker/__init__.py
@@ -0,0 +1,10 @@
+#  Copyright 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+#  with the License. A copy of the License is located at
+#
+#  http://aws.amazon.com/apache2.0/
+#
+#  or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES
+#  OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions
+#  and limitations under the License.
diff --git a/test/pecos/xmr/test_reranker.py b/test/pecos/xmr/test_reranker.py
@@ -0,0 +1,19 @@
+#  Copyright 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
+#  with the License. A copy of the License is located at
+#
+#  http://aws.amazon.com/apache2.0/
+#
+#  or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES
+#  OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions
+#  and limitations under the License.
+import pytest  # noqa: F401; pylint: disable=unused-variable
+
+
+def test_importable():
+    import pecos.xmr  # noqa: F401
+    import pecos.xmr.reranker  # noqa: F401
+    from pecos.xmr.reranker.model import TextNumrEncoder  # noqa: F401
+    from pecos.xmr.reranker.model import RankingModel  # noqa: F401
+    from pecos.xmr.reranker.trainer import RankingTrainer  # noqa: F401