Spaces:
Runtime error
Runtime error
fix ulteriori per CI
Browse files- CI/test_ci.py +4 -0
- requirements.txt +2 -2
CI/test_ci.py
CHANGED
|
@@ -1,5 +1,9 @@
|
|
| 1 |
# Utilities
|
|
|
|
| 2 |
import torch
|
|
|
|
|
|
|
|
|
|
| 3 |
from src.modello import Modello
|
| 4 |
from src.dataset import LoadDataset
|
| 5 |
from sklearn.metrics import accuracy_score
|
|
|
|
| 1 |
# Utilities
|
| 2 |
+
import sys
|
| 3 |
import torch
|
| 4 |
+
sys.modules['torch'] = torch
|
| 5 |
+
import os
|
| 6 |
+
os.environ["CUDA_VISIBLE_DEVICES"] = ""
|
| 7 |
from src.modello import Modello
|
| 8 |
from src.dataset import LoadDataset
|
| 9 |
from sklearn.metrics import accuracy_score
|
requirements.txt
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
--extra-index-url https://download.pytorch.org/whl/cpu
|
| 2 |
-
transformers
|
| 3 |
torch==2.2.1+cpu
|
| 4 |
-
|
|
|
|
| 5 |
pandas
|
| 6 |
scikit-learn
|
| 7 |
datasets
|
|
|
|
| 1 |
--extra-index-url https://download.pytorch.org/whl/cpu
|
|
|
|
| 2 |
torch==2.2.1+cpu
|
| 3 |
+
numpy==1.26.4
|
| 4 |
+
transformers==4.38.2
|
| 5 |
pandas
|
| 6 |
scikit-learn
|
| 7 |
datasets
|