From 933063f766391363b2635a7f5f8e32e07996b16c Mon Sep 17 00:00:00 2001
From: Guillermo Rodriguez <guillermo@telos.net>
Date: Mon, 3 Feb 2025 17:00:44 -0300
Subject: [PATCH] Fix some import related TODOs

---
 skynet/dgpu/compute.py |  7 -------
 skynet/dgpu/daemon.py  |  7 +++----
 skynet/dgpu/network.py | 10 +---------
 3 files changed, 4 insertions(+), 20 deletions(-)

diff --git a/skynet/dgpu/compute.py b/skynet/dgpu/compute.py
index 6b2f092..f069e60 100755
--- a/skynet/dgpu/compute.py
+++ b/skynet/dgpu/compute.py
@@ -7,17 +7,10 @@ import gc
 import logging
 
 from hashlib import sha256
-# import zipfile
-# from PIL import Image
-# from diffusers import DiffusionPipeline
 
 import trio
 import torch
 
-# from skynet.constants import (
-#     DEFAULT_INITAL_MODEL,
-#     MODELS,
-# )
 from skynet.dgpu.errors import (
     DGPUComputeError,
     DGPUInferenceCancelled,
diff --git a/skynet/dgpu/daemon.py b/skynet/dgpu/daemon.py
index c95e1ca..0f49ff7 100755
--- a/skynet/dgpu/daemon.py
+++ b/skynet/dgpu/daemon.py
@@ -1,11 +1,10 @@
-from datetime import datetime
-from functools import partial
-from hashlib import sha256
 import json
 import logging
 import random
-# import traceback
 import time
+from datetime import datetime
+from functools import partial
+from hashlib import sha256
 
 import trio
 from quart import jsonify
diff --git a/skynet/dgpu/network.py b/skynet/dgpu/network.py
index 654cec8..21ddfd8 100755
--- a/skynet/dgpu/network.py
+++ b/skynet/dgpu/network.py
@@ -2,7 +2,6 @@ import io
 import json
 import time
 import logging
-
 from pathlib import Path
 from functools import partial
 
@@ -11,12 +10,7 @@ import leap
 import anyio
 import httpx
 import outcome
-
-from PIL import (
-    Image,
-    # UnidentifiedImageError,  # TODO, remove?
-)
-
+from PIL import Image
 from leap.cleos import CLEOS
 from leap.protocol import Asset
 from skynet.constants import (
@@ -28,8 +22,6 @@ from skynet.ipfs import (
     AsyncIPFSHTTP,
     get_ipfs_file,
 )
-# TODO, remove?
-# from skynet.dgpu.errors import DGPUComputeError
 
 
 REQUEST_UPDATE_TIME: int = 3