Hugging Face's logo

Spaces:
cleanify-ai
/
Data-cleaning
Sleeping

App Files Community
thorodin103 commited on
Commit
7d721e6
·
verified ·
1 Parent(s): d417287

Upload environment.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. environment.py +0 -2
environment.py CHANGED
@@ -60,8 +60,6 @@ class DataCleaningEnv:
60
  folder = mapping.get(self.task_id, "easy")
61
  self.current_df = pd.read_csv(f"datasets/{folder}/dirty.csv")
62
  self.gold_df = pd.read_csv(f"datasets/{folder}/gold.csv")
63
- # Apply random variations so agent cannot memorize
64
- self.current_df = self._add_random_variation(self.current_df)
65
 
66
  def _add_random_variation(self, df: pd.DataFrame) -> pd.DataFrame:
67
  """Add small random variations to prevent memorization."""
 
60
  folder = mapping.get(self.task_id, "easy")
61
  self.current_df = pd.read_csv(f"datasets/{folder}/dirty.csv")
62
  self.gold_df = pd.read_csv(f"datasets/{folder}/gold.csv")
 
 
63
 
64
  def _add_random_variation(self, df: pd.DataFrame) -> pd.DataFrame:
65
  """Add small random variations to prevent memorization."""