Multi-line examples can be tested + ALBERT patch for CircleCI
All tests should now work fine.
This commit is contained in:
@@ -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):
|
|||||||
)
|
)
|
||||||
|
|
||||||
# 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 @@
|
|||||||
# 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):
|
|||||||
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):
|
|||||||
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]
|
||||||
|
|||||||
Reference in New Issue
Block a user