From 7c21cb456c6f3f5f9610a26620286ef67e115518 Mon Sep 17 00:00:00 2001
From: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com>
Date: Tue, 29 Nov 2022 18:20:23 +0530
Subject: [PATCH] Update __init__.py

---
 src/pet/__init__.py | 9 +--------
 1 file changed, 1 insertion(+), 8 deletions(-)

diff --git a/src/pet/__init__.py b/src/pet/__init__.py
index 77a32cc..51ab10c 100644
--- a/src/pet/__init__.py
+++ b/src/pet/__init__.py
@@ -4,14 +4,7 @@
 
 __version__ = "0.1.0.dev0"
 
-from .pet_model import (
-    PETModel,
-    PETModelForCausalLM,
-    PETModelForSeq2SeqLM,
-    PETModelForSequenceClassification,
-    PETPluginBase,
-    PETType,
-)
+from .pet_model import PETModel, PETModelForCausalLM, PETModelForSeq2SeqLM, PETModelForSequenceClassification
 from .task_mapping import MODEL_TYPE_TO_PROMPT_MODEL_MAPPING
 from .tuners import (
     PrefixEncoder,