From e7add45241e1aa0a0e3706979abbbb02632e981d Mon Sep 17 00:00:00 2001
From: Anyang Peng <137014849+anyangml@users.noreply.github.com>
Date: Sun, 14 Jul 2024 06:36:49 +0800
Subject: [PATCH] Fix: tf dipole atomic key (#3975)

I believe this should fix #3970

<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->

## Summary by CodeRabbit

- **Bug Fixes**
- Updated label naming convention from "atomic_" to "atom_" to ensure
consistency in labeling.

<!-- end of auto-generated comment: release notes by coderabbit.ai -->
---
 deepmd/tf/loss/tensor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deepmd/tf/loss/tensor.py b/deepmd/tf/loss/tensor.py
index d3a1b95369..4a70ae2a96 100644
--- a/deepmd/tf/loss/tensor.py
+++ b/deepmd/tf/loss/tensor.py
@@ -148,7 +148,7 @@ def label_requirement(self) -> List[DataRequirementItem]:
         # data required
         data_requirements.append(
             DataRequirementItem(
-                "atomic_" + self.label_name,
+                "atom_" + self.label_name,
                 self.tensor_size,
                 atomic=True,
                 must=False,