optimization/__init__.py

from .ActivationExtractor import ActivationExtractor
from .optimize_activation_batch import optimize_activation_batch