Support M1 Mac for MLMD OSS. #143

Update Bazel settings to support both macos and macos_arm64.
Use the bazel_skylib library for config selects.
Reflect changes in setup.py.

PiperOrigin-RevId: 432116433

Author: BrianSong

.bazelrc

@@ -24,3 +24,5 @@ build --incompatible_restrict_string_escapes=false
 # parameter 'user_link_flags' is deprecated and will be removed soon.
 # It may be temporarily re-enabled by setting --incompatible_require_linker_input_cc_api=false
 build --incompatible_require_linker_input_cc_api=false
+build:macos --apple_platform_type=macos
+build:macos_arm64 --cpu=darwin_arm64

ml_metadata/BUILD

@@ -1,5 +1,7 @@
 licenses(["notice"])  # Apache 2.0
 
+load("@bazel_skylib//lib:selects.bzl", "selects")
+
 package(default_visibility = ["//visibility:public"])
 
 config_setting(
@@ -10,9 +12,26 @@ config_setting(
 )
 
 config_setting(
+    name = "macos_x86_64",
+    values = {
+        "apple_platform_type": "macos",
+        "cpu": "darwin",
+    },
+)
+
+config_setting(
+    name = "macos_arm64",
+    values = {
+        "apple_platform_type": "macos",
+        "cpu": "darwin_arm64",
+    },
+)
+
+selects.config_setting_group(
     name = "macos",
-    constraint_values = [
-        "@bazel_tools//platforms:osx",
+    match_any = [
+        ":macos_x86_64",
+        ":macos_arm64",
     ],
 )
 

setup.py

@@ -91,8 +91,12 @@ def finalize_options(self):
           'installation instruction.')
     self._additional_build_options = []
     if platform.system() == 'Darwin':
-      # Minimum macOS version 10.14 is needed to be compatible with c++17.
-      self._additional_build_options = ['--macos_minimum_os=10.14']
+      # This flag determines the platform qualifier of the macos wheel.
+      if platform.machine() == 'arm64':
+        self._additional_build_options = ['--macos_minimum_os=11.0',
+                                          '--config=macos_arm64']
+      else:
+        self._additional_build_options = ['--macos_minimum_os=10.14']
 
   def run(self):
     subprocess.check_call(