Fix some import related TODOs

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,

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

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