🐛 Finalize, nbt import

src/craftground/environment/environment.py

@@ -6,6 +6,7 @@
 from enum import Enum
 import threading
 from typing import Tuple, Optional, TypedDict, Union, List, Any, Dict
+import weakref
 
 import gymnasium as gym
 import numpy as np
@@ -144,6 +145,8 @@ def __init__(
             self.render_action,
         )
 
+        weakref.finalize(self, self.close)
+
     def reset(
         self,
         *,

src/craftground/nbt/models/entity.py

@@ -1,6 +1,6 @@
 from dataclasses import dataclass
 from typing import Optional
-from nbt_dataclass import (
+from ..nbt_dataclass import (
     NBTBase,
     NBTByte,
     NBTCompound,

src/craftground/nbt/models/structure.py

@@ -1,8 +1,8 @@
 from __future__ import annotations
 from dataclasses import dataclass, field
 from typing import Optional
-from models.entity import EntityNBT
-from nbt_dataclass import (
+from ..models.entity import EntityNBT
+from ..nbt_dataclass import (
     NBTBase,
     NBTCompound,
     NBTDouble,

src/craftground/nbt/nbt_io.py

@@ -3,7 +3,7 @@
 from typing import Tuple
 
 import numpy as np
-from nbt_dataclass import (
+from .nbt_dataclass import (
     NBTBase,
     NBTByte,
     NBTByteArray,

src/craftground/nbt/structure_editor.py

@@ -1,8 +1,8 @@
 from typing import List, Optional, Dict, Tuple
-from models.structure import BlockNBT, PaletteNBT, StructureEntityNBT, StructureNBT
-from models.entity import EntityNBT, ItemEntityNBT, ItemNBT
-from nbt_dataclass import NBTCompound, NBTInt, NBTShort, NBTString
-from nbt_io import read_nbt, write_nbt
+from .models.structure import BlockNBT, PaletteNBT, StructureEntityNBT, StructureNBT
+from .models.entity import ItemEntityNBT, ItemNBT
+from .nbt_dataclass import NBTCompound, NBTInt, NBTShort, NBTString
+from .nbt_io import read_nbt, write_nbt
 
 
 class Structure: