Daniel Gil-U Fuhge commited on
Commit
c7db08f
1 Parent(s): 310c2df

update preprocessing

Browse files
Files changed (1) hide show
  1. src/preprocessing/preprocessing.py +1 -1
src/preprocessing/preprocessing.py CHANGED
@@ -1,7 +1,7 @@
1
  # Imports
2
  import os
3
  import copy
4
- import torch
5
  import glob
6
  import pandas as pd
7
  import pickle
 
1
  # Imports
2
  import os
3
  import copy
4
+ import torch.nn as torch
5
  import glob
6
  import pandas as pd
7
  import pickle