Update all_actions.py

Amber-MD · Jun 17, 2024 · 45e1d50 · 45e1d50
1 parent 38f468b
commit 45e1d50
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytraj/all_actions.py b/pytraj/all_actions.py
@@ -1,6 +1,6 @@
 from __future__ import absolute_import
 import numpy as np
-from enum import Enum
+from enum import StrEnum
 from typing import Any, Callable, List, Union
 from functools import partial
 
@@ -71,7 +71,7 @@
     'watershell', 'wavelet', 'xcorr', 'xtalsymm',
 ] # yapf: disable
 
-class DatasetType(Enum):
+class DatasetType(StrEnum):
     COORDS = 'coords'
     REFERENCE = 'reference'
     REFERENCE_FRAME = 'ref_frame'