🐛 Fixed TypeError type is not subscriptable in datasets

[marmg]

Status	Type	⚠️ Core Change	Issue
Ready	Bug	No	Link

Problem

It seems next typing fails with Python < 3.9:
def load_medmentions() -> DatasetDict[DatasetWithEntities]:

This is due to the class DatasetDict of datasets extending dict. dict is a type, and it's not subscriptable in Python<3.9 (for typing it's used Dict, not dict)

See Related StackOverflow post

Solution

Now, instead of return typing DatasetDict[DatasetWithEntities] is Dict[Union[str, Split], DatasetWithEntities]

Closes #3

[codecov]

Codecov Report

Base: 91.72% // Head: 91.73% // Increases project coverage by +0.00% 🎉

Coverage data is based on head (c6194cd) compared to base (79c1d29).
Patch coverage: 100.00% of modified lines in pull request are covered.

Additional details and impacted files

@@           Coverage Diff           @@
##             main      #28   +/-   ##
=======================================
  Coverage   91.72%   91.73%           
=======================================
  Files          53       53           
  Lines        2030     2032    +2     
=======================================
+ Hits         1862     1864    +2     
  Misses        168      168           

Impacted Files	Coverage Δ
...ot/evaluation/dataset/med_mentions/med_mentions.py	100.00% <100.00%> (ø)
zshot/evaluation/dataset/ontonotes/onto_notes.py	64.58% <100.00%> (+0.75%)	⬆️

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.

☔ View full report at Codecov.
📢 Do you have feedback about the report comment? Let us know in this issue.