调整mattergen-wrapper位置

mars_toolkit/compute/material_gen.py

@@ -28,14 +28,7 @@ from mars_toolkit.core.config import config
 import sys
 import os
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../../')))
-from mattergen_wrapper import generator
-CrystalGenerator = generator.CrystalGenerator
-from mattergen.common.data.types import TargetProperty
-from mattergen.common.utils.eval_utils import MatterGenCheckpointInfo
-from mattergen.common.utils.data_classes import (
-    PRETRAINED_MODEL_NAME,
-    MatterGenCheckpointInfo,
-)
+from ..core.mattergen_wrapper import *
 
 logger = logging.getLogger(__name__)