NotImplementedError: numpy() is only available when eager execution is enabled