Project import generated by Copybara.

GitOrigin-RevId: 6e4aff1cc351be3ae4537b677f36d139ee50ce09
2021-03-25 22:09:18 -04:00
parent a92cff7a60
commit 7c331ad58b
175 changed files with 4804 additions and 1325 deletions
@@ -180,13 +180,13 @@ class GeneratePyProtos(setuptools.Command):
        'mediapipe/util/**/*.proto'
    ]:
      for proto_file in glob.glob(pattern, recursive=True):
+        proto_dir = os.path.dirname(os.path.abspath(proto_file))
        # Ignore test protos.
        if proto_file.endswith('test.proto'):
          continue
-        # Ignore tensorflow protos.
-        if 'tensorflow' in proto_file:
+        # Ignore tensorflow protos in mediapipe/calculators/tensorflow.
+        if 'tensorflow' in proto_dir:
          continue
-        proto_dir = os.path.dirname(os.path.abspath(proto_file))
        # Ignore testdata dir.
        if proto_dir.endswith('testdata'):
          continue