old_contents: # Copyright 2019 DeepMind Technologies Ltd. All rights reserved.
  #
  # Licensed under the Apache License, Version 2.0 (the "License");
  # you may not use this file except in compliance with the License.
  # You may obtain a copy of the License at
  #
  #     http://www.apache.org/licenses/LICENSE-2.0
  #
  # Unless required by applicable law or agreed to in writing, software
  # distributed under the License is distributed on an "AS IS" BASIS,
  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  # See the License for the specific language governing permissions and
  # limitations under the License.
  
  # Copyright 2019 DeepMind Technologies Ltd. All rights reserved.
  #
  # Licensed under the Apache License, Version 2.0 (the "License");
  # you may not use this file except in compliance with the License.
  # You may obtain a copy of the License at
  #
  #     http://www.apache.org/licenses/LICENSE-2.0
  #
  # Unless required by applicable law or agreed to in writing, software
  # distributed under the License is distributed on an "AS IS" BASIS,
  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  # See the License for the specific language governing permissions and
  # limitations under the License.
  #
  # Lint as: python3
  """An integration test building and testing open_spiel wheel."""
  import os
  import sys
  import sysconfig
  
  import nox
  
  # Note: versions are maintained here and in scripts/travis_script.sh.
  PYTHON_JAX_DEPS = "jax==0.2.7 jaxlib==0.1.57 dm-haiku==0.0.3 optax==0.0.2 chex==0.0.3"
  PYTHON_PYTORCH_DEPS = "torch==1.7.0"
  PYTHON_TENSORFLOW_DEPS = "tensorflow==2.4.1 tensorflow-probability<0.8.0,>=0.7.0"
  
  def get_distutils_tempdir():
    return (
        f"temp.{sysconfig.get_platform()}-{sys.version_info[0]}.{sys.version_info[1]}"
    )
  
  
  @nox.session(python="3")
  def tests(session):
    session.install("-r", "requirements.txt")
    child_env = os.environ.copy()
    child_env["OPEN_SPIEL_BUILD_ALL"] = "ON"
    if child_env["OPEN_SPIEL_ENABLE_JAX"] == "ON":
      for dep in PYTHON_JAX_DEPS.split():
        session.install(dep)
    if child_env["OPEN_SPIEL_ENABLE_PYTORCH"] == "ON":
      for dep in PYTHON_PYTORCH_DEPS.split()
        session.install(dep)
    if child_env["OPEN_SPIEL_ENABLE_TENSORFLOW"] == "ON":
      for dep in PYTHON_TENSORFLOW_DEPS.split()
        session.install(dep)
    session.run("python3", "setup.py", "build", env=child_env)
    session.run("python3", "setup.py", "install", env=child_env)
    session.cd(os.path.join("build", get_distutils_tempdir()))
    session.run(
        "ctest", f"-j{4*os.cpu_count()}", "--output-on-failure", external=True)
  
new_contents: # Copyright 2019 DeepMind Technologies Ltd. All rights reserved.
  #
  # Licensed under the Apache License, Version 2.0 (the "License");
  # you may not use this file except in compliance with the License.
  # You may obtain a copy of the License at
  #
  #     http://www.apache.org/licenses/LICENSE-2.0
  #
  # Unless required by applicable law or agreed to in writing, software
  # distributed under the License is distributed on an "AS IS" BASIS,
  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  # See the License for the specific language governing permissions and
  # limitations under the License.
  
  # Copyright 2019 DeepMind Technologies Ltd. All rights reserved.
  #
  # Licensed under the Apache License, Version 2.0 (the "License");
  # you may not use this file except in compliance with the License.
  # You may obtain a copy of the License at
  #
  #     http://www.apache.org/licenses/LICENSE-2.0
  #
  # Unless required by applicable law or agreed to in writing, software
  # distributed under the License is distributed on an "AS IS" BASIS,
  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  # See the License for the specific language governing permissions and
  # limitations under the License.
  #
  # Lint as: python3
  """An integration test building and testing open_spiel wheel."""
  import os
  import sys
  import sysconfig
  
  import nox
  
  # Note: versions are maintained here and in scripts/travis_script.sh.
  PYTHON_JAX_DEPS = "jax==0.2.7 jaxlib==0.1.57 dm-haiku==0.0.3 optax==0.0.2 chex==0.0.3"
  PYTHON_PYTORCH_DEPS = "torch==1.7.0"
  PYTHON_TENSORFLOW_DEPS = "tensorflow==2.4.1 tensorflow-probability<0.8.0,>=0.7.0"
  
  def get_distutils_tempdir():
    return (
        f"temp.{sysconfig.get_platform()}-{sys.version_info[0]}.{sys.version_info[1]}"
    )
  
  
  @nox.session(python="3")
  def tests(session):
    session.install("-r", "requirements.txt")
    child_env = os.environ.copy()
    child_env["OPEN_SPIEL_BUILD_ALL"] = "ON"
    if child_env["OPEN_SPIEL_ENABLE_JAX"] == "ON":
      for dep in PYTHON_JAX_DEPS.split():
        session.install(dep)
    if child_env["OPEN_SPIEL_ENABLE_PYTORCH"] == "ON":
      for dep in PYTHON_PYTORCH_DEPS.split():
        session.install(dep)
    if child_env["OPEN_SPIEL_ENABLE_TENSORFLOW"] == "ON":
      for dep in PYTHON_TENSORFLOW_DEPS.split():
        session.install(dep)
    session.run("python3", "setup.py", "build", env=child_env)
    session.run("python3", "setup.py", "install", env=child_env)
    session.cd(os.path.join("build", get_distutils_tempdir()))
    session.run(
        "ctest", f"-j{4*os.cpu_count()}", "--output-on-failure", external=True)
  
subject: Add missing colons