Skip to content
Permalink
Browse files

Fix package imports

  • Loading branch information...
achyudh committed Mar 21, 2019
1 parent 01a00d1 commit 6f3ce0456e591c6a630c5bd7fe87e999f4c5ff26
Showing with 3 additions and 4 deletions.
  1. +2 −2 __init__.py
  2. +0 −1 models/xml_cnn/args.py
  3. +1 −1 setup.py
@@ -1,3 +1,3 @@
from models import reg_lstm, kim_cnn
from models import reg_lstm, kim_cnn, han, char_cnn, xml_cnn

__all__ = ['reg_lstm', 'kim_cnn']
__all__ = ['reg_lstm', 'kim_cnn', 'char_cnn', 'xml_cnn', 'han']
@@ -24,6 +24,5 @@ def get_args():
parser.add_argument('--resume-snapshot', type=str)
parser.add_argument('--trained-model', type=str)


args = parser.parse_args()
return args
@@ -3,5 +3,5 @@
setup(name='hedwig',
version='1.0.0',
description='PyTorch deep learning models for document classification',
packages=['char_cnn', 'conv_rnn', 'kim_cnn', 'reg_lstm', 'xml_cnn'],
packages=['char_cnn', 'han', 'kim_cnn', 'reg_lstm', 'xml_cnn'],
)

0 comments on commit 6f3ce04

Please sign in to comment.
You can’t perform that action at this time.