From 63c2b1a39d0699fd10aad9fc359b14cabeb90e9e Mon Sep 17 00:00:00 2001 From: Renan Alves de Oliveira Date: Sat, 6 Jun 2020 15:57:01 -0300 Subject: [PATCH] add sys library --- map2map/test.py | 1 + 1 file changed, 1 insertion(+) diff --git a/map2map/test.py b/map2map/test.py index 1fc193c..a3609ee 100644 --- a/map2map/test.py +++ b/map2map/test.py @@ -1,6 +1,7 @@ from pprint import pprint import numpy as np import torch +import sys from torch.utils.data import DataLoader from .data import FieldDataset