From cbbb3c43c55d2d93a156fc80bd12f31ecbac8520 Mon Sep 17 00:00:00 2001
From: Julien Chaumond <chaumond@gmail.com>
Date: Thu, 23 Apr 2020 16:27:43 -0400
Subject: [PATCH] [hubconf] Modify pythonpath to get canonical imports to work

See https://github.com/huggingface/transformers/pull/3881/files#r412292660

Should we remove SRC_DIR from sys.path right after the imports, @aaugustin?
---
 hubconf.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/hubconf.py b/hubconf.py
index b473d41c14..cadad632f1 100644
--- a/hubconf.py
+++ b/hubconf.py
@@ -1,12 +1,19 @@
-from src.transformers import (
+import os
+import sys
+
+SRC_DIR = os.path.join(os.path.dirname(__file__), "src")
+sys.path.append(SRC_DIR)
+
+
+from transformers import (
     AutoConfig,
     AutoModel,
     AutoModelForQuestionAnswering,
     AutoModelForSequenceClassification,
     AutoModelWithLMHead,
     AutoTokenizer,
+    add_start_docstrings,
 )
-from src.transformers.file_utils import add_start_docstrings
 
 
 dependencies = ["torch", "numpy", "tokenizers", "boto3", "filelock", "requests", "tqdm", "regex", "sentencepiece", "sacremoses"]