vikenkd commited on
Commit
c3d09d7
·
verified ·
1 Parent(s): 32b7369

Delete tests

Browse files
Files changed (2) hide show
  1. tests/data_ingestion.py +0 -17
  2. tests/test_gpu.py +0 -6
tests/data_ingestion.py DELETED
@@ -1,17 +0,0 @@
1
- from src.config import CatDogDatasetConfigsInput
2
- from src.data_ingestion import CatDogDataset
3
- def main():
4
- # Initialize data ingestion with configurations
5
- data_configs = CatDogDatasetConfigsInput(
6
- data_path="datasets/datasets.zip",
7
- train_data_path="datasets/train",
8
- test_data_path="datasets/test",
9
- test_size=0.2,
10
- random_state=42
11
- )
12
- print(data_configs)
13
- dataset = CatDogDataset(data_configs)
14
- dataset.load_data()
15
-
16
- if __name__ == "__main__":
17
- main()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
tests/test_gpu.py DELETED
@@ -1,6 +0,0 @@
1
- import torch
2
-
3
- if __name__ == "__main__":
4
- print(torch.cuda.is_available())
5
- print(torch.version.cuda)
6
- print(torch.cuda.get_device_name(0) if torch.cuda.is_available() else "No GPU detected")