dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix test

This commit is contained in:
thomwolf 2019-10-11 15:55:01 +02:00
parent 0f9fc4fbde
commit 1f5d9513d8
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
transformers/tests/modeling_tf_xlnet_test.py
View File

@ -161,7 +161,8 @@ class TFXLNetModelTest(TFCommonTestCases.TFCommonModelTester):
"outputs": outputs.numpy(),
}
model.config.mem_len = 0
config.mem_len = 0
model = TFXLNetModel(config)
no_mems_outputs = model(inputs)
self.parent.assertEqual(len(no_mems_outputs), 1)

4
transformers/tests/modeling_xlnet_test.py
View File

@ -150,7 +150,9 @@ class XLNetModelTest(CommonTestCases.CommonModelTester):
"outputs": outputs,
}
model.config.mem_len = 0
config.mem_len = 0
model = XLNetModel(config)
model.eval()
no_mems_outputs = model(input_ids_1)
self.parent.assertEqual(len(no_mems_outputs), 1)