Readd aims magnetism test

Author: tpurcell90

src/atomate2/aims/schemas/calculation.py

@@ -7,10 +7,11 @@
 from collections.abc import Sequence
 from datetime import datetime, timezone
 from pathlib import Path
-from typing import TYPE_CHECKING, Any, Optional, Union
+from typing import Any, Optional, Union
 
 import numpy as np
 from ase.spectrum.band_structure import BandStructure
+from emmet.core.math import Matrix3D, Vector3D
 from jobflow.utils import ValueEnum
 from pydantic import BaseModel, Field
 from pymatgen.core import Molecule, Structure
@@ -21,9 +22,6 @@
 from pymatgen.io.common import VolumetricData
 from typing_extensions import Self
 
-if TYPE_CHECKING:
-    from emmet.core.math import Matrix3D, Vector3D
-
 STORE_VOLUMETRIC_DATA = ("total_density",)
 
 

tests/aims/test_flows/test_magnetism.py

@@ -15,12 +15,6 @@
 cwd = os.getcwd()
 
 
-@pytest.mark.skip(
-    reason="pymatgen 2024.11.13 broke this test with ValueError: Structure contains "
-    "magnetic moments on both magmom site properties and spin species properties. This "
-    "is ambiguous. Remove one or the other."
-)
-# TODO re-attempt to fix and unskip this test
 def test_magnetic_orderings(mock_aims, tmp_path, species_dir, mg2mn4o8):
     parameters = {
         "k_grid": [2, 2, 2],