feat(tf): implement change-bias command

.gitignore

@@ -51,7 +51,12 @@ buildcxx/
 node_modules/
 *.bib.original
 
+# Coverage files
+.coverage
+.coverage.*
+
 # Test output files (temporary)
 test_dp_test/
 test_dp_test_*.out
 *_detail.out
+out.json

deepmd/main.py

@@ -721,12 +721,13 @@ def main_parser() -> argparse.ArgumentParser:
     parser_change_bias = subparsers.add_parser(
         "change-bias",
         parents=[parser_log],
-        help="(Supported backend: PyTorch) Change model out bias according to the input data.",
+        help="Change model out bias according to the input data.",
         formatter_class=RawTextArgumentDefaultsHelpFormatter,
         epilog=textwrap.dedent(
             """\
         examples:
-            dp change-bias model.pt -s data -n 10 -m change
+            dp --pt change-bias model.pt -s data -n 10 -m change
+            dp --tf change-bias model.ckpt -s data -n 10 -m change
         """
         ),
     )

deepmd/tf/entrypoints/__init__.py

@@ -4,6 +4,9 @@
 from ..infer.model_devi import (
     make_model_devi,
 )
+from .change_bias import (
+    change_bias,
+)
 from .compress import (
     compress,
 )
@@ -34,6 +37,7 @@
 )
 
 __all__ = [
+    "change_bias",
     "compress",
     "convert",
     "doc_train_input",