From 6a1f995290a62df31d42b9a1f270a2a67cad0206 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Thu, 14 Mar 2024 06:25:06 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 deepmd/tf/fit/dos.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/deepmd/tf/fit/dos.py b/deepmd/tf/fit/dos.py
index fe7572524f..aef134da92 100644
--- a/deepmd/tf/fit/dos.py
+++ b/deepmd/tf/fit/dos.py
@@ -562,9 +562,7 @@ def build(
 
             outs = tf.reshape(
                 final_layer,
-                [
-                    tf.shape(inputs)[0], natoms[0], self.numb_dos
-                ],
+                [tf.shape(inputs)[0], natoms[0], self.numb_dos],
             )
         # add bias
         # self.atom_ener_before = outs