From 7cc72373672e754c8f539e1aa4b653381e2219a8 Mon Sep 17 00:00:00 2001
From: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Date: Mon, 15 Nov 2021 20:29:31 -0500
Subject: [PATCH] fix ImportError; add a check for it

---
 .github/workflows/test_import.py | 18 ++++++++++++++++++
 dpdata/pymatgen/molecule.py      |  5 ++++-
 2 files changed, 22 insertions(+), 1 deletion(-)
 create mode 100644 .github/workflows/test_import.py

diff --git a/.github/workflows/test_import.py b/.github/workflows/test_import.py
new file mode 100644
index 000000000..69eaf2464
--- /dev/null
+++ b/.github/workflows/test_import.py
@@ -0,0 +1,18 @@
+name: test Python import
+
+on:
+  - push
+  - pull_request
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v2
+    - uses: actions/setup-python@v2
+        with:
+            python-version: '3.x'
+            architecture: 'x64'
+    - run: python -m pip install .
+    - run: python -c 'import dpdata'
+
diff --git a/dpdata/pymatgen/molecule.py b/dpdata/pymatgen/molecule.py
index f671d6ea8..06b74c168 100644
--- a/dpdata/pymatgen/molecule.py
+++ b/dpdata/pymatgen/molecule.py
@@ -1,5 +1,8 @@
 import numpy as np
-from pymatgen.core import Molecule
+try:
+    from pymatgen.core import Molecule
+except ImportError:
+    pass
 from collections import Counter
 import dpdata