From b132723661a34b3370810071520c8801ea943134 Mon Sep 17 00:00:00 2001
From: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Date: Sun, 26 Dec 2021 20:53:34 -0500
Subject: [PATCH] fix NameError

---
 deepmd/entrypoints/freeze.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/deepmd/entrypoints/freeze.py b/deepmd/entrypoints/freeze.py
index 8b0acf0e98..172c765646 100755
--- a/deepmd/entrypoints/freeze.py
+++ b/deepmd/entrypoints/freeze.py
@@ -7,6 +7,7 @@
 """
 
 import logging
+import google.protobuf.message
 from deepmd.env import tf, FITTING_NET_PATTERN
 from deepmd.utils.sess import run_sess
 from deepmd.utils.graph import get_pattern_nodes_from_graph_def