fixing import error

Author: HojeChun

nff/io/chgnet.py

@@ -1,6 +1,6 @@
 """Convert NFF Dataset to CHGNet StructureData"""
 
-from typing import Dict, List
+from typing import Dict, List, TYPE_CHECKING
 import functools
 import random
 import torch
@@ -12,7 +12,7 @@
 from nff.io import AtomsBatch
 from nff.utils.cuda import batch_detach, detach
 from chgnet.graph import CrystalGraph, CrystalGraphConverter
-from typing import TYPE_CHECKING
+
 if TYPE_CHECKING:
     from collections.abc import Sequence