From 52154267a7769a32e4d6369260618d62739d4181 Mon Sep 17 00:00:00 2001
From: Myungchul Keum <8174871+dofuuz@users.noreply.github.com>
Date: Thu, 20 Feb 2025 15:10:34 +0900
Subject: [PATCH] Fix 'ceil' is not defined

---
 vector_quantize_pytorch/residual_fsq.py | 2 +-
 vector_quantize_pytorch/residual_lfq.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/vector_quantize_pytorch/residual_fsq.py b/vector_quantize_pytorch/residual_fsq.py
index 82753a3..6c19310 100644
--- a/vector_quantize_pytorch/residual_fsq.py
+++ b/vector_quantize_pytorch/residual_fsq.py
@@ -1,5 +1,5 @@
 import random
-from math import log2
+from math import ceil
 from functools import partial
 
 from typing import List
diff --git a/vector_quantize_pytorch/residual_lfq.py b/vector_quantize_pytorch/residual_lfq.py
index 475025b..bf4c3f0 100644
--- a/vector_quantize_pytorch/residual_lfq.py
+++ b/vector_quantize_pytorch/residual_lfq.py
@@ -1,5 +1,5 @@
 import random
-from math import log2
+from math import ceil, log2
 from functools import partial, cache
 
 import torch