From 27c24a5f8fc03e1c7fe39b93ec5e39bc1d35ecdf Mon Sep 17 00:00:00 2001
From: Direk Kakkar <millioniar1717@gmail.com>
Date: Sat, 18 Jul 2026 21:24:54 +0530
Subject: [PATCH] fix: removed extra import

---
 tabfm/src/jax/model.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/tabfm/src/jax/model.py b/tabfm/src/jax/model.py
index 42f3019..7c6a7e4 100644
--- a/tabfm/src/jax/model.py
+++ b/tabfm/src/jax/model.py
@@ -32,7 +32,6 @@
 Y: Number of layers (a.k.a blocks)
 """
 
-from absl import logging
 import chex
 import math
 from math import pi