fix: missing numpy import to abstract_leaf.py

braun-steven · Nov 11, 2024 · c4ea3d0 · c4ea3d0
1 parent 8cabee2
commit c4ea3d0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/simple_einet/layers/distributions/abstract_leaf.py b/simple_einet/layers/distributions/abstract_leaf.py
@@ -1,6 +1,7 @@
 import logging
 from abc import ABC, abstractmethod
 from typing import List
+import numpy as np
 
 import torch
 from torch import distributions as dist, nn