Problème de typing dict vs Dict

[picaultj]

Bonjour,

Je rencontre un petit problème.
Lors de l'import de: from vigogne.preprocess import generate_instruct_prompt, l'erreur suivante apparait (Linux, Python 3.8.13):

...
  File "/home/jerome/miniconda2/envs/origami/lib/python3.8/site-packages/vigogne/data_utils.py", line 39, in Conversation
    def fully_model_dump(self, **kwargs) -> dict[str, Any]:
TypeError: 'type' object is not subscriptable

dans data_utils.py, remplacer la ligne 39:
def fully_model_dump(self, **kwargs) -> dict[str, Any]:

avec

def fully_model_dump(self, **kwargs) -> Dict[str, Any]:

et rajout de from typing import Dict permet de résoudre mon problème

Serait-il possible de corriger le code?
Merci! (et bravo pour ce superbe boulot)

[bofenghuang]

Bonjour @picaultj, merci d'avoir repéré le problème !