From af7f9046f0317c9d0c332677cd457cbb9b9a9af6 Mon Sep 17 00:00:00 2001 From: PRG Admin Date: Fri, 5 Feb 2021 12:12:33 -0800 Subject: [PATCH] fix import --- scripts/attention.py | 8 ++++---- scripts/sample.py | 8 ++++---- transvae/version.py | 2 +- 3 files changed, 9 insertions(+), 9 deletions(-) diff --git a/scripts/attention.py b/scripts/attention.py index 057a04a..f422ae8 100644 --- 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 index 7d0dfc8..0659e41 100644 --- 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 index 16aa1c8..e2b8033 100644 --- 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