mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
60d4d5e1aa
* Remove all __future__ imports from RLlib. * Remove (object) again from tf_run_builder.py::TFRunBuilder. * Fix 2xLINT warnings. * Fix broken appo_policy import (must be appo_tf_policy) * Remove future imports from all other ray files (not just RLlib). * Remove future imports from all other ray files (not just RLlib). * Remove future import blocks that contain `unicode_literals` as well. Revert appo_tf_policy.py to appo_policy.py (belongs to another PR). * Add two empty lines before Schedule class. * Put back __future__ imports into determine_tests_to_run.py. Fails otherwise on a py2/print related error.
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
import os
|
|
import sys
|
|
"""
|
|
This script is used for modifying the generated java flatbuffer
|
|
files for the reason: The package declaration in Java is different
|
|
from python and C++, and there is no option in the flatc command
|
|
to specify package(namepsace) for Java specially.
|
|
|
|
USAGE:
|
|
python modify_generated_java_flatbuffers_file.py RAY_HOME
|
|
|
|
RAY_HOME: The root directory of Ray project.
|
|
"""
|
|
|
|
# constants declarations
|
|
PACKAGE_DECLARATION = "package org.ray.runtime.generated;"
|
|
|
|
|
|
def add_package(file):
|
|
with open(file, "r") as file_handler:
|
|
lines = file_handler.readlines()
|
|
|
|
if "FlatBuffers" not in lines[0]:
|
|
return
|
|
|
|
lines.insert(1, PACKAGE_DECLARATION + os.linesep)
|
|
with open(file, "w") as file_handler:
|
|
for line in lines:
|
|
file_handler.write(line)
|
|
|
|
|
|
def add_package_declarations(generated_root_path):
|
|
file_names = os.listdir(generated_root_path)
|
|
for file_name in file_names:
|
|
if not file_name.endswith(".java"):
|
|
continue
|
|
full_name = os.path.join(generated_root_path, file_name)
|
|
add_package(full_name)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
ray_home = sys.argv[1]
|
|
root_path = os.path.join(
|
|
ray_home, "java/runtime/src/main/java/org/ray/runtime/generated")
|
|
add_package_declarations(root_path)
|