From e8ed2e799f47c8b0d364ec2dbee6a98005c8ebdf Mon Sep 17 00:00:00 2001 From: Wei-Cheng Chang Date: Thu, 17 Oct 2024 14:10:02 -0700 Subject: [PATCH] Make pecos.xmr importable and add unit test (#301) Co-authored-by: Wei-Cheng Chang --- pecos/xmr/__init__.py | 10 ++++++++++ pecos/xmr/reranker/__init__.py | 10 ++++++++++ test/pecos/xmr/test_reranker.py | 34 +++++++++++++++++++++++++++++++++ 3 files changed, 54 insertions(+) create mode 100644 pecos/xmr/__init__.py create mode 100644 pecos/xmr/reranker/__init__.py create mode 100644 test/pecos/xmr/test_reranker.py diff --git a/pecos/xmr/__init__.py b/pecos/xmr/__init__.py new file mode 100644 index 00000000..695dd79a --- /dev/null +++ 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 new file mode 100644 index 00000000..695dd79a --- /dev/null +++ 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 new file mode 100644 index 00000000..60c79bc3 --- /dev/null +++ b/test/pecos/xmr/test_reranker.py @@ -0,0 +1,34 @@ +# 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 + + +def test_model(): + from pecos.xmr.reranker.model import NumrMLPEncoderConfig + + mlp_config = NumrMLPEncoderConfig( + inp_feat_dim=5, + inp_dropout_prob=0.5, + hid_actv_type="gelu", + hid_size_list=[8, 16], + ) + assert mlp_config.inp_feat_dim == 5 + assert mlp_config.inp_dropout_prob == 0.5 + assert mlp_config.hid_actv_type == "gelu" + assert mlp_config.hid_size_list == [8, 16]