upd dependencies

Author: loribonna

datasets/bias_celeba_utils/celeba.py

@@ -5,9 +5,7 @@
 import logging
 import os
 from collections import namedtuple
-from typing import Any, Callable, List, Optional, Tuple, Union
-import numpy as np
-import pandas as pd
+from typing import Any, Callable, Optional, Tuple
 from PIL import Image
 import torch
 from torchvision.datasets.utils import check_integrity, download_file_from_google_drive, extract_archive, verify_str_arg
@@ -45,6 +43,7 @@ def __init__(
         version: int = 1,
     ) -> None:
         super().__init__(root, transform=transform, target_transform=target_transform)
+        import pandas as pd
 
         self.split = split
         self.target_type = "attr"

datasets/seq_eurosat_rgb.py

@@ -2,9 +2,7 @@
 import json
 import logging
 import os
-import sys
 import zipfile
-import pandas as pd
 import requests
 import torch
 import torchvision.transforms as transforms
@@ -30,6 +28,7 @@ class MyEuroSat(Dataset):
 
     def __init__(self, root, split='train', transform=None,
                  target_transform=None) -> None:
+        import pandas as pd
 
         self.root = root
         self.split = split
@@ -65,6 +64,7 @@ def __init__(self, root, split='train', transform=None,
 
     @staticmethod
     def get_class_names():
+        import pandas as pd
         if not os.path.exists(base_path() + f'eurosat/DONE'):
             gdd.download_file_from_google_drive(file_id='1Ip7yaCWFi0eaOFUGga0lUdVi_DDQth1o',
                                                 dest_path=base_path() + 'eurosat/split.json')

pyproject.toml

@@ -6,10 +6,12 @@ dependencies = [
     "accelerate>=1.6.0",
     "bitsandbytes>=0.45.4",
     "ftfy>=6.3.1",
+    "googledrivedownloader==0.4",
     "gpustat>=1.1.1",
     "kornia>=0.7.0",
     "numpy>=2.2.4",
     "onedrivedownloader>=1.1.3",
+    "pandas>=2.2.3",
     "pillow>=11.1.0",
     "pyyaml>=6.0.2",
     "regex>=2024.11.6",