Bug fix: forgot that some models were actually using sparsemax, replaced patch on hparams.yaml (#244)

Author: Ricardo Rei

comet/models/__init__.py

@@ -101,7 +101,7 @@ def load_from_checkpoint(
             import pkg_resources
             comet_version = pkg_resources.get_distribution("unbabel-comet").version
             use_softmax = (pkg_resources.parse_version(comet_version) >= pkg_resources.parse_version("2.2.4") and 
-                          hparams.get("layer_transformation") == "sparsemax")
+                          hparams.get("layer_transformation") == "sparsemax_patch")
         except:
             use_softmax = False
 

comet/modules/layerwise_attention.py

@@ -48,15 +48,6 @@ def __init__(
 
         self.transform_fn = torch.softmax
         if layer_transformation == "sparsemax":
-            # Import warnings module
-            import warnings
-            # Display warning message
-            warnings.warn(
-                "WARNING - sparsemax is DEPRECATED in favor of softmax. "
-                "Please use softmax instead. ",
-                DeprecationWarning,
-                stacklevel=2
-            )
             from entmax import sparsemax
             self.transform_fn = sparsemax