fix import

oriondollar · Feb 5, 2021 · af7f904 · af7f904
1 parent 6a0fb8d
commit af7f904
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git a/scripts/attention.py b/scripts/attention.py
@@ -8,11 +8,11 @@
 import torch
 from torch.autograd import Variable
 
-from TransVAE.transvae.trans_models import TransVAE
-from TransVAE.transvae.rnn_models import RNN, RNNAttn
+from transvae.trans_models import TransVAE
+from transvae.rnn_models import RNN, RNNAttn
 
-from TransVAE.transvae.data import vae_data_gen, make_std_mask
-from TransVAE.scripts.parsers import attn_parser
+from transvae.data import vae_data_gen, make_std_mask
+from scripts.parsers import attn_parser
 
 def calc_attention(args):
     ### Load model

diff --git a/scripts/sample.py b/scripts/sample.py
@@ -5,10 +5,10 @@
 import numpy as np
 import pandas as pd
 
-from TransVAE.transvae.trans_models import TransVAE
-from TransVAE.transvae.rnn_models import RNN, RNNAttn
-from TransVAE.transvae.tvae_util import calc_entropy
-from TransVAE.scripts.parsers import sample_parser
+from transvae.trans_models import TransVAE
+from transvae.rnn_models import RNN, RNNAttn
+from transvae.tvae_util import calc_entropy
+from scripts.parsers import sample_parser
 
 def sample(args):
     ### Load model

diff --git a/transvae/version.py b/transvae/version.py
@@ -4,7 +4,7 @@
 # Format expected by setup.py and doc/source/conf.py: string of form "X.Y.Z"
 _version_major = 0
 _version_minor = 4
-_version_micro = 2  # use '' for first of series, number for 1 and above
+_version_micro = 3  # use '' for first of series, number for 1 and above
 # _version_extra = 'dev1'
 _version_extra = ''  # Uncomment this for full releases