From 81b5949e16a2c62dd733154673af418713d6803d Mon Sep 17 00:00:00 2001
From: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Date: Fri, 17 May 2024 23:14:43 -0400
Subject: [PATCH] style: enable W rules (#3793)

Fix #3789.

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

## Summary by CodeRabbit

- **Documentation**
- Corrected a minor formatting issue in the quick start guide for
displaying a file path.

- **Chores**
  - Added a new linting rule for warnings to improve code quality.

- **Tests**
- Improved regular expression handling in test cases by using raw string
notation.

<!-- end of auto-generated comment: release notes by coderabbit.ai -->

---------

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
---
 doc/getting-started/quick_start.ipynb     | 2 +-
 pyproject.toml                            | 1 +
 source/tests/pt/model/test_fitting_net.py | 2 +-
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/doc/getting-started/quick_start.ipynb b/doc/getting-started/quick_start.ipynb
index 67674c4654..67462c91d4 100644
--- a/doc/getting-started/quick_start.ipynb
+++ b/doc/getting-started/quick_start.ipynb
@@ -396,7 +396,7 @@
     }
    ],
    "source": [
-    "! cat DeePMD-kit_Tutorial/00.data/training_data/type.raw "
+    "! cat DeePMD-kit_Tutorial/00.data/training_data/type.raw"
    ]
   },
   {
diff --git a/pyproject.toml b/pyproject.toml
index 23d42e73d2..80d5ad9ee9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -298,6 +298,7 @@ docstring-code-format = true
 [tool.ruff.lint]
 select = [
     "E", # errors
+    "W", # warning
     "F", # pyflakes
     "D", # pydocstyle
     "UP", # pyupgrade
diff --git a/source/tests/pt/model/test_fitting_net.py b/source/tests/pt/model/test_fitting_net.py
index c7e1723799..900b01d687 100644
--- a/source/tests/pt/model/test_fitting_net.py
+++ b/source/tests/pt/model/test_fitting_net.py
@@ -113,7 +113,7 @@ def test_consistency(self):
         ).to(env.DEVICE)
         for name, param in my_fn.named_parameters():
             matched = re.match(
-                "filter_layers\.networks\.(\d).layers\.(\d)\.([a-z]+)", name
+                r"filter_layers\.networks\.(\d).layers\.(\d)\.([a-z]+)", name
             )
             key = None
             if matched: