Minor fixes

src/editor/operations/get_colour_lut.ipynb

@@ -17,11 +17,11 @@
     }
    ],
    "source": [
+    "import sys\n",
+    "sys.path.insert(0, \"../../\")\n",
+    "\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
-    "import sys\n",
-    "\n",
-    "sys.path.insert(0, \"../../\")\n",
     "from editor.operations import get_random_saturation_per_hue_lut\n",
     "\n",
     "\n",
@@ -85,7 +85,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.2"
+   "version": "3.1.-1"
   }
  },
  "nbformat": 4,

src/editor/utils/__init__.py

@@ -2,3 +2,4 @@ from .random import random
 from .compute_histogram import compute_histogram
 from .generate_rotation_matrices import generate_rotation_matrices
 from .get_next_run_name import get_next_run_name
+from .kldiv import kldiv

src/editor/utils/kldiv.py

@@ -0,0 +1,11 @@
+import numpy as np
+
+
+def kldiv(P: np.ndarray, Q: np.ndarray) -> float:
+    P /= P.sum()
+    Q /= Q.sum()
+
+    P_safe = np.maximum(P, np.finfo(float).eps)
+    Q_safe = np.maximum(Q, np.finfo(float).eps)
+
+    return np.sum(P_safe * np.log(P_safe / Q_safe))