Roboproch commited on
Commit
559e539
·
1 Parent(s): fd14b9d

fix ulteriori per CI

Browse files
Files changed (2) hide show
  1. CI/test_ci.py +4 -0
  2. 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
- torchvision==0.17.1+cpu
 
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