From 2d5eb33e0463633d54d29e64da85329fbaca8d1d Mon Sep 17 00:00:00 2001
From: QuanyiLi <quanyili0057@gmail.com>
Date: Sat, 6 May 2023 23:32:16 +0100
Subject: [PATCH] suppres tf warning

---
 scenarionet/converter/waymo/utils.py  | 7 +++++--
 scenarionet/examples/convert_waymo.py | 4 +---
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/scenarionet/converter/waymo/utils.py b/scenarionet/converter/waymo/utils.py
index c2e8bba..96b9fff 100644
--- a/scenarionet/converter/waymo/utils.py
+++ b/scenarionet/converter/waymo/utils.py
@@ -9,11 +9,14 @@ logger = logging.getLogger(__name__)
 import numpy as np
 
 try:
+    os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'  # FATAL
+    logging.getLogger('tensorflow').setLevel(logging.FATAL)
     import tensorflow as tf
+
 except ImportError as e:
     logger.info(e)
 try:
-    from waymo_open_dataset.protos.scenario_pb2 import Scenario
+    from waymo_open_dataset.protos import scenario_pb2
 except ImportError as e:
     logger.warning(e, "\n Please install waymo_open_dataset package: pip install waymo-open-dataset-tf-2-11-0==1.5.0")
 
@@ -429,7 +432,7 @@ def get_waymo_scenarios(waymo_data_direction):
         if ("tfrecord" not in file_path) or (not os.path.isfile(file_path)):
             continue
         for data in tf.data.TFRecordDataset(file_path, compression_type="").as_numpy_iterator():
-            scenario = Scenario()
+            scenario = scenario_pb2.Scenario()
             scenario.ParseFromString(data)
             # a trick for loging file name
             scenario.scenario_id = scenario.scenario_id + SPLIT_KEY + file
diff --git a/scenarionet/examples/convert_waymo.py b/scenarionet/examples/convert_waymo.py
index 9640c43..1e20a41 100644
--- a/scenarionet/examples/convert_waymo.py
+++ b/scenarionet/examples/convert_waymo.py
@@ -1,13 +1,11 @@
 import logging
 import os
-
+from scenarionet import SCENARIONET_DATASET_PATH
 from scenarionet.converter.utils import write_to_directory
 from scenarionet.converter.waymo.utils import convert_waymo_scenario, get_waymo_scenarios
 
 logger = logging.getLogger(__name__)
 
-from scenarionet import SCENARIONET_DATASET_PATH
-
 if __name__ == '__main__':
     force_overwrite = True
     dataset_name = "waymo"