Only check for `HaikuFramework` if haiku is loaded (!1132) · Merge requests · Vicentini Filippo / netket

Vicentini Filippo requested to merge framework into master Mar 08, 2022

Created by: femtobit

Currently, using the JaxFramework model framework without haiku being installed causes an exception, because HaikuFramework.is_my_module (which tries to import haiku) is called anyways.

This PR fixes this by only checking for haiku-based modules if haiku is already loaded.

Only check for `HaikuFramework` if haiku is loaded

Merge request reports