further fix windows

sail-sg · Dec 6, 2023 · 183c96e · 183c96e
1 parent 8e79848
commit 183c96e
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/gen_repo/gen_py.py b/gen_repo/gen_py.py
@@ -87,10 +87,10 @@ def main(_):
 
     functionals.append((p, ext_params, ext_params_descriptions, info))
 
-  with open(FLAGS.template, "rt") as f:
+  with open(FLAGS.template, "rt", encoding="utf8") as f:
     py_template = Template(f.read(), trim_blocks=True, lstrip_blocks=True)
     py_code = py_template.render(functionals=functionals, zip=zip)
-    with open(FLAGS.output, "wt") as out:
+    with open(FLAGS.output, "wt", encoding="utf8") as out:
       out.write(py_code)
 
 

diff --git a/gen_repo/libxc/build.jinja b/gen_repo/libxc/build.jinja
@@ -110,6 +110,7 @@ genrule(
     srcs = [":libxc.so"],
     outs = ["libxc.pyd"],
     cmd = "cp $< $@",
+    visibility = ["//visibility:public"],
 )
 
 py_library(

diff --git a/gen_repo/wheel.BUILD b/gen_repo/wheel.BUILD
@@ -68,6 +68,8 @@ py_wheel(
         "@jax_xc_repo//jax_xc:utils",
         "@jax_xc_repo//jax_xc/impl",
         "@jax_xc_repo//jax_xc/libxc",
-        "@jax_xc_repo//jax_xc/libxc:libxc.so",
-    ],
+    ] + select({
+        "@platforms//os:windows": ["@jax_xc_repo//jax_xc/libxc:gen_pyd"],
+        "//conditions:default": ["@jax_xc_repo//jax_xc/libxc:libxc.so"],
+    }),
 )