search

rs239 / ablm

Protein language model customized for antibodies
MIT License
60 stars 7 forks source link

Type Checking for Task should be str, not int #9

Open samsledje opened 6 months ago

samsledje commented 6 months ago

https://github.com/rs239/ablm/blob/9b891b3c71f1b10d4c15dd40b58f7cfcc003e13e/abmap/commands/embed.py#L29

samsledje commented 6 months ago

Choices = ['structure', 'function']

samsledje commented 6 months ago

Likewise in L66 type=int should be changed to type=str

samsledje commented 6 months ago

L153 should be changed to task_ = task passed as argument