Commit 64abd3e0 authored by Lysandre's avatar Lysandre Committed by Lysandre Debut
Browse files

Multi-line examples can be tested + ALBERT patch for CircleCI

All tests should now work fine.
parent 83757725
...@@ -79,6 +79,7 @@ class AlbertConfig(PretrainedConfig): ...@@ -79,6 +79,7 @@ class AlbertConfig(PretrainedConfig):
Example:: Example::
from transformers import AlbertConfig, AlbertModel
# Initializing an ALBERT-xxlarge style configuration # Initializing an ALBERT-xxlarge style configuration
albert_xxlarge_configuration = AlbertConfig() albert_xxlarge_configuration = AlbertConfig()
...@@ -90,7 +91,7 @@ class AlbertConfig(PretrainedConfig): ...@@ -90,7 +91,7 @@ class AlbertConfig(PretrainedConfig):
) )
# Initializing a model from the ALBERT-base style configuration # Initializing a model from the ALBERT-base style configuration
model = AlbertModel(bert_base_configuration) model = AlbertModel(albert_xxlarge_configuration)
# Accessing the model configuration # Accessing the model configuration
configuration = model.config configuration = model.config
......
...@@ -13,8 +13,8 @@ ...@@ -13,8 +13,8 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
import unittest
import os import os
import unittest
def get_examples_from_file(file): def get_examples_from_file(file):
...@@ -37,7 +37,7 @@ def get_examples_from_file(file): ...@@ -37,7 +37,7 @@ def get_examples_from_file(file):
example_mode = True example_mode = True
example_indentation = line.lower().find("example::") example_indentation = line.lower().find("example::")
return examples return ['\n'.join(example) for example in examples]
class TestCodeExamples(unittest.TestCase): class TestCodeExamples(unittest.TestCase):
...@@ -51,8 +51,7 @@ class TestCodeExamples(unittest.TestCase): ...@@ -51,8 +51,7 @@ class TestCodeExamples(unittest.TestCase):
print("Testing", configuration_file, str(len(examples)) + "/" + str(len(examples))) print("Testing", configuration_file, str(len(examples)) + "/" + str(len(examples)))
def execute_example(code_example): def execute_example(code_example):
for line in code_example: exec(code_example)
exec(line)
with self.subTest(msg=configuration_file): with self.subTest(msg=configuration_file):
[execute_example(code_example) for code_example in examples] [execute_example(code_example) for code_example in examples]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment