-- changed .tmp to pt_tmp

-- unfroze scikit learn version

Author: manujosephv

pytorch_tabular/config/config.py

@@ -479,22 +479,22 @@ def read_from_yaml(filename: str = "config/optimizer_config.yml"):
 class ExperimentRunManager:
     def __init__(
         self,
-        exp_version_manager: str = ".tmp/exp_version_manager.yml",
+        exp_version_manager: str = ".pt_tmp/exp_version_manager.yml",
     ) -> None:
         """The manages the versions of the experiments based on the name. It is a simple dictionary(yaml) based lookup.
         Primary purpose is to avoid overwriting of saved models while runing the training without changing the experiment name.
 
         Args:
             exp_version_manager (str, optional): The path of the yml file which acts as version control.
-            Defaults to ".tmp/exp_version_manager.yml".
+            Defaults to ".pt_tmp/exp_version_manager.yml".
         """
         super().__init__()
         self._exp_version_manager = exp_version_manager
         if os.path.exists(exp_version_manager):
             self.exp_version_manager = OmegaConf.load(exp_version_manager)
         else:
             self.exp_version_manager = OmegaConf.create({})
-            os.makedirs(".tmp", exist_ok=True)
+            os.makedirs(".pt_tmp", exist_ok=True)
             with open(self._exp_version_manager, "w") as file:
                 OmegaConf.save(config=self.exp_version_manager, f=file)
 

requirements.txt

@@ -2,7 +2,7 @@ torch>=1.4
 category-encoders==2.2.2
 numpy>=1.17.2
 pandas==1.1.5
-scikit-learn==0.23.2
+scikit-learn>=0.23.2
 pytorch-lightning==1.3.6
 omegaconf>=2.0.1
 torchmetrics>=0.3.2