From 66b2e683166f5df2c7afde8fede2bf1e789fa638 Mon Sep 17 00:00:00 2001
From: Konstantin Nikolaou <konsti.nikolaou@gmail.com>
Date: Thu, 15 Aug 2024 17:32:41 +0200
Subject: [PATCH] add new observables to init

---
 papyrus/measurements/__init__.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/papyrus/measurements/__init__.py b/papyrus/measurements/__init__.py
index ffdd395..92d80b7 100644
--- a/papyrus/measurements/__init__.py
+++ b/papyrus/measurements/__init__.py
@@ -25,9 +25,11 @@
 from papyrus.measurements.measurements import (
     NTK,
     Accuracy,
+    LabelNTKAlignment,
     Loss,
     LossDerivative,
     NTKEigenvalues,
+    NTKEigenvectorAlignment,
     NTKEntropy,
     NTKMagnitudeDistribution,
     NTKSelfEntropy,
@@ -45,4 +47,6 @@
     Accuracy.__name__,
     NTK.__name__,
     LossDerivative.__name__,
+    LabelNTKAlignment.__name__,
+    NTKEigenvectorAlignment.__name__,
 ]