Spaces:
Sleeping
Sleeping
Delete tests
Browse files- tests/data_ingestion.py +0 -17
- 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")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|