Update: Automatic config json write & dataset download in pkg init

ChrisIsKing · May 20, 2023 · cb3f07e · cb3f07e
1 parent 3ba8626
commit cb3f07e
Show file tree

Hide file tree

Showing 5 changed files with 382 additions and 368 deletions.
diff --git a/setup.py b/setup.py
@@ -1,6 +1,6 @@
 from setuptools import setup, find_packages
 
-VERSION = '0.1.1'
+VERSION = '0.1.2'
 DESCRIPTION = """
 code and data for the Findings of ACL'23 paper Label Agnostic Pre-training for Zero-shot Text Classification 
 by Christopher Clarke, Yuzhao Heng, Yiping Kang, Krisztian Flautner, Lingjia Tang and Jason Mars
@@ -15,7 +15,7 @@
     description=DESCRIPTION,
     long_description=DESCRIPTION,
     url='https://github.com/ChrisIsKing/zero-shot-text-classification',
-    download_url='https://github.com/ChrisIsKing/zero-shot-text-classification/archive/refs/tags/v0.1.0.tar.gz',
+    download_url='https://github.com/ChrisIsKing/zero-shot-text-classification/archive/refs/tags/v0.1.2.tar.gz',
     packages=find_packages(),
     include_package_data=True,
     install_requires=[

diff --git a/zeroshot_classifier/models/__init__.py b/zeroshot_classifier/models/__init__.py
@@ -1,5 +1,3 @@
 from . architecture import *
-from . import dual_bi_encoder
 from .gpt2 import *
 from . import gpt3
-from . import gpt_neo