Use self._checkpoint instead of checkpoint in LocalGlobalLocalBlock

NASA-IMPACT / Prithvi-WxC

Implementation of the Prithvi WxC Foundation Model and Downstream Tasks

MIT License

98 stars 13 forks source link

Use self._checkpoint instead of checkpoint in LocalGlobalLocalBlock #9

Closed Jubeku closed 1 month ago

Jubeku commented 1 month ago

If checkpoint is used, its default value of None causes an error as the code expects an iterable list. self._checkpoint instead is set to [] in case of checkpoint == None.

WillTrojak commented 1 month ago

Looks alright, I'll fix the pep8 line limit