diff --git a/bedms/attr_standardizer.py b/bedms/attr_standardizer.py index fbcaf39..0ddc6e0 100644 --- a/bedms/attr_standardizer.py +++ b/bedms/attr_standardizer.py @@ -87,7 +87,6 @@ def _get_parameters(self) -> Tuple[int, int, int, int, int, float]: def _load_model(self) -> nn.Module: """ Calls function to load the model from HuggingFace repository and sets to eval(). - :return nn.Module: Loaded Neural Network Model. """ try: diff --git a/bedms/utils.py b/bedms/utils.py index 632c1d9..0b1415c 100644 --- a/bedms/utils.py +++ b/bedms/utils.py @@ -202,7 +202,11 @@ def data_encoding( # print(transformed_columns) X_values_bow = transformed_columns # Label Encoding +<<<<<<< HEAD:bedms/utils.py label_encoder =None +======= + label_encoder = None +>>>>>>> b13f0a3 (minor changes):attribute_standardizer/utils.py lb_path = hf_hub_download( repo_id=REPO_ID, filename=ENCODE_LABEL_ENCODER_FILENAME,