From eacded568ba82ffda58744347d6e9b953dfcd96c Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Fri, 13 Jun 2025 06:56:46 -0400
Subject: [PATCH] Fix ruff.

---
 comfy/ldm/cosmos/predict2.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/comfy/ldm/cosmos/predict2.py b/comfy/ldm/cosmos/predict2.py
index 89a80ca4f..3b91b3f6e 100644
--- a/comfy/ldm/cosmos/predict2.py
+++ b/comfy/ldm/cosmos/predict2.py
@@ -5,9 +5,8 @@ from torch import nn
 from einops import rearrange
 from einops.layers.torch import Rearrange
 import logging
-from typing import Any, Callable, List, Optional, Tuple
+from typing import Callable, Optional, Tuple
 import math
-import numpy as np
 
 from .position_embedding import VideoRopePosition3DEmb, LearnablePosEmbAxis
 from torchvision import transforms