minor tweak

molecularnodes/io/parse/cellpack.py

@@ -1,6 +1,5 @@
 from pathlib import Path
 from typing import Union
-from pathlib import Path
 import numpy as np
 import bpy
 
@@ -22,7 +21,7 @@ def __init__(self, file_path: Union[str, Path]) -> None:
         self.chain_ids = self.data.chain_ids
 
     def create_model(
-        self, name: str = "StarFileObject", node_setup: bool = True, world_scale: float = 0.01
+        self, name: str = "StarFileObject", node_setup: bool = True, world_scale: float = 0.01, fraction: float = 1.0
     ) -> bpy.types.Object:
         self.data_object = self._create_data_object(name=f"{name}")
         self._create_object_instances(name=name, node_setup=node_setup)

molecularnodes/io/parse/ensemble.py

@@ -1,7 +1,7 @@
 import bpy
 from abc import ABCMeta
 import numpy as np
-from typing import Union, Optional, Set, List, Dict
+from typing import Union
 from pathlib import Path
 from ... import blender as bl
 import warnings
@@ -26,7 +26,7 @@ def __init__(self, file_path: Union[str, Path]) -> None:
 
     @classmethod
     def create_model(
-        self, name: str = "EnsembleObject", node_setup: bool = True, world_scale: float = 0.01
+        self, name: str = "EnsembleObject", node_setup: bool = True, world_scale: float = 0.01, fraction: float = 1.0
     ) -> bpy.types.Object:
         """
         Create a 3D model in the of the ensemble.

molecularnodes/io/parse/star.py

@@ -2,8 +2,7 @@
 import bpy
 import starfile.typing
 from bpy.app.handlers import persistent
-from typing import Union, Optional, Set, List, Dict
-from numpy.typing import NDArray
+from typing import Union, Optional, Dict
 from pathlib import Path
 import starfile
 from .ensemble import Ensemble
@@ -201,7 +200,7 @@ def _update_micrograph_texture(self, *_) -> None:
             self.star_node.inputs["Micrograph"].default_value = image_obj
 
     def create_model(
-        self, name: str = "EnsembleObject", node_setup: bool = True, world_scale: float = 0.01
+        self, name: str = "EnsembleObject", node_setup: bool = True, world_scale: float = 0.01, fraction: float = 1.0
     ) -> bpy.types.Object:
         from molecularnodes.blender.nodes import get_star_node, MN_micrograph_material
 

molecularnodes/ui/ops.py

@@ -1,6 +1,6 @@
 import bpy
 from ..blender import nodes
-from typing import Set, Union, Optional, Dict
+from typing import Set
 
 
 class MN_OT_Add_Custom_Node_Group(bpy.types.Operator):