From c0a31606deaa0505a21fb23438f25c2800d3700f Mon Sep 17 00:00:00 2001
From: Sam Hadow <sam.hadow@proton.me>
Date: Mon, 20 Apr 2026 12:08:51 +0200
Subject: [PATCH] run script

---
 run_sage.sh           | 2 ++
 src/tea3/tea3model.py | 3 +++
 2 files changed, 5 insertions(+)
 create mode 100755 run_sage.sh

diff --git a/run_sage.sh b/run_sage.sh
new file mode 100755
index 0000000..9eeadc7
--- /dev/null
+++ b/run_sage.sh
@@ -0,0 +1,2 @@
+#!/bin/bash
+PYTHONPATH=src sage src/tea3/tea3model.py
diff --git a/src/tea3/tea3model.py b/src/tea3/tea3model.py
index 95d86f8..b472cf0 100644
--- a/src/tea3/tea3model.py
+++ b/src/tea3/tea3model.py
@@ -1,5 +1,8 @@
 from sage.all import *
 
+from tea3.constants import TEA3_SBOX, T_F1, T_F2
+
+
 F = GF(2)
 
 names = (