I think so, but sometimes appears extrange errors like np is not defined or partial is not defined.
I will try it a little more to understand what it is happening
edit: it only happens in my notebook, it should be related to the corrupted conda installation. In the office it works smooth