【MIIT program】Add GMTNet for dielectric tensor prediction

MANIFEST.in

@@ -0,0 +1,5 @@
+global-exclude *.pdparams
+global-exclude *.pkl
+prune miit_logs
+prune paddle_scatter
+prune jointContribution

ppmat/datasets/__init__.py

@@ -31,6 +31,7 @@
 
 from ppmat.datasets import collate_fn
 from ppmat.datasets.high_level_water_dataset import HighLevelWaterDataset
+from ppmat.datasets.gmtnet_dataset import GMTNetDielectricDataset
 from ppmat.datasets.jarvis_dataset import JarvisDataset
 from ppmat.datasets.matbench_dataset import MatbenchDataset
 from ppmat.datasets.mp20_dataset import AlexMP20MatterGenDataset
@@ -62,6 +63,7 @@
     "MPTrjDataset",
     "JarvisDataset",
     "HighLevelWaterDataset",
+    "GMTNetDielectricDataset",
     "MSDnmrDataset",
     "MatbenchDataset",
     "DensityDataset", 

ppmat/datasets/collate_fn.py

@@ -88,6 +88,18 @@ def __call__(self, batch: List[Any]) -> Any:
         )
 
 
+class GMTNetDielectricCollate:
+    """Collate GMTNet dielectric samples with batch size one."""
+
+    def __init__(self):
+        pass
+
+    def __call__(self, batch):
+        from ppmat.datasets.gmtnet_dataset import gmtnet_dielectric_collate_fn
+
+        return gmtnet_dielectric_collate_fn(batch)
+
+
 class DensityCollator:
     def __init__(
         self,