2 Commits

Author SHA1 Message Date
Vladimir Blagojevic
72168eddaf
Add BatchEncoding flatten (#1562)
* Add BatchEncoding flatten

* Rename BatchEncoding flatten to flatten_rename

* Unit test for BatchEncoding flatten_rename
2021-10-07 15:29:57 +02:00
Timo Moeller
537204e8c9
Fix tests and adjust folder structure
* Add type annotations in QuestionAnsweringHead

* Fix test by increasing max_seq_len

* Add SampleBasket type annotation

* Remove prediction head param from adaptive model init

* Add type ignore for AdaptiveModel init

* Fix and rename tests

* Adjust folder structure

Co-authored-by: Julian Risch <julian.risch@deepset.ai>
2021-09-13 18:38:14 +02:00