Fix lint errors in freeze_graph.py

tensorflow · Oct 17, 2018 · aca8771 · aca8771
1 parent b24d17c
commit aca8771
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/freeze_graph.py b/freeze_graph.py
@@ -11,6 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+"""Freeze a model to a GraphDef proto."""
 
 from absl import app, flags
 
@@ -21,7 +22,8 @@
 FLAGS = flags.FLAGS
 
 
-def main(argv):
+def main(unused_argv):
+    """Freeze a model to a GraphDef proto."""
     if FLAGS.use_tpu:
         dual_net.freeze_graph_tpu(FLAGS.model_path)
     else: