From 1fcf82cffc19136c69640ffbfd534ec32ab8ce0e Mon Sep 17 00:00:00 2001
From: root <2000011006@stu.pku.edu.cn>
Date: Fri, 13 Dec 2024 21:42:04 +0800
Subject: [PATCH] Solve pre-commit

---
 deepmd/pt/model/task/property.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/deepmd/pt/model/task/property.py b/deepmd/pt/model/task/property.py
index c5d0285caf..233cd3ddb2 100644
--- a/deepmd/pt/model/task/property.py
+++ b/deepmd/pt/model/task/property.py
@@ -2,6 +2,7 @@
 import logging
 from typing import (
     Optional,
+    Union
 )
 
 import torch