search

bofenghuang / vigogne

French instruction-following and chat models
Apache License 2.0
497 stars 47 forks source link

Problème de typing dict vs Dict #29

Open picaultj opened 1 year ago

picaultj commented 1 year ago

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 commented 1 year ago

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