From a9347c6713b0ca75c046fe341968017d9a5eb69c Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Thu, 16 Jan 2025 14:09:37 -0800
Subject: [PATCH] Fix pylint error

---
 comfy/utils.py   | 4 ++--
 requirements.txt | 3 ++-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/comfy/utils.py b/comfy/utils.py
index 0dccea7f0..84d1d0c9d 100644
--- a/comfy/utils.py
+++ b/comfy/utils.py
@@ -57,9 +57,9 @@ if hasattr(torch.serialization, "add_safe_globals"):  # TODO: this was added in
 
     ModelCheckpoint.__module__ = "pytorch_lightning.callbacks.model_checkpoint"
 
-    from numpy.core.multiarray import scalar
+    from numpy.core.multiarray import scalar  # pylint: disable=no-name-in-module
     from numpy import dtype
-    from numpy.dtypes import Float64DType
+    from numpy.dtypes import Float64DType  # pylint: disable=no-name-in-module
     from _codecs import encode
 
     torch.serialization.add_safe_globals([ModelCheckpoint, scalar, dtype, Float64DType, encode])
diff --git a/requirements.txt b/requirements.txt
index bc32316d4..9c38346c3 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -69,4 +69,5 @@ skia-python
 pebble>=5.0.7
 openai
 anthropic
-humanize
\ No newline at end of file
+humanize
+lightning
\ No newline at end of file