ax-pytest-plugin-debugger.ipynb

ax-pytest-plugin-debugger.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": [],
      "authorship_tag": "ABX9TyNq2uuW7HbaFmZKOFnQ5WUm",
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/gist/sgbaird/834d141c886f298f6d323505ae51a284/ax-pytest-plugin-debugger.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "%pip install ax-platform"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "uZuVOdwXLcgK",
        "outputId": "348fa814-81af-48ad-baf7-7a78fd14d519"
      },
      "execution_count": 13,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Requirement already satisfied: ax-platform in /usr/local/lib/python3.10/dist-packages (0.3.4)\n",
            "Requirement already satisfied: botorch==0.9.2 in /usr/local/lib/python3.10/dist-packages (from ax-platform) (0.9.2)\n",
            "Requirement already satisfied: jinja2 in /usr/local/lib/python3.10/dist-packages (from ax-platform) (3.1.2)\n",
            "Requirement already satisfied: pandas in /usr/local/lib/python3.10/dist-packages (from ax-platform) (1.5.3)\n",
            "Requirement already satisfied: scipy in /usr/local/lib/python3.10/dist-packages (from ax-platform) (1.10.1)\n",
            "Requirement already satisfied: scikit-learn in /usr/local/lib/python3.10/dist-packages (from ax-platform) (1.2.2)\n",
            "Requirement already satisfied: ipywidgets in /usr/local/lib/python3.10/dist-packages (from ax-platform) (7.7.1)\n",
            "Requirement already satisfied: plotly>=5.12.0 in /usr/local/lib/python3.10/dist-packages (from ax-platform) (5.15.0)\n",
            "Requirement already satisfied: typeguard==2.13.3 in /usr/local/lib/python3.10/dist-packages (from ax-platform) (2.13.3)\n",
            "Requirement already satisfied: multipledispatch in /usr/local/lib/python3.10/dist-packages (from botorch==0.9.2->ax-platform) (1.0.0)\n",
            "Requirement already satisfied: torch>=1.13.1 in /usr/local/lib/python3.10/dist-packages (from botorch==0.9.2->ax-platform) (2.0.1+cu118)\n",
            "Requirement already satisfied: pyro-ppl>=1.8.4 in /usr/local/lib/python3.10/dist-packages (from botorch==0.9.2->ax-platform) (1.8.6)\n",
            "Requirement already satisfied: gpytorch==1.11 in /usr/local/lib/python3.10/dist-packages (from botorch==0.9.2->ax-platform) (1.11)\n",
            "Requirement already satisfied: linear-operator==0.5.1 in /usr/local/lib/python3.10/dist-packages (from botorch==0.9.2->ax-platform) (0.5.1)\n",
            "Requirement already satisfied: jaxtyping>=0.2.9 in /usr/local/lib/python3.10/dist-packages (from linear-operator==0.5.1->botorch==0.9.2->ax-platform) (0.2.20)\n",
            "Requirement already satisfied: tenacity>=6.2.0 in /usr/local/lib/python3.10/dist-packages (from plotly>=5.12.0->ax-platform) (8.2.2)\n",
            "Requirement already satisfied: packaging in /usr/local/lib/python3.10/dist-packages (from plotly>=5.12.0->ax-platform) (23.1)\n",
            "Requirement already satisfied: ipykernel>=4.5.1 in /usr/local/lib/python3.10/dist-packages (from ipywidgets->ax-platform) (5.5.6)\n",
            "Requirement already satisfied: ipython-genutils~=0.2.0 in /usr/local/lib/python3.10/dist-packages (from ipywidgets->ax-platform) (0.2.0)\n",
            "Requirement already satisfied: traitlets>=4.3.1 in /usr/local/lib/python3.10/dist-packages (from ipywidgets->ax-platform) (5.7.1)\n",
            "Requirement already satisfied: widgetsnbextension~=3.6.0 in /usr/local/lib/python3.10/dist-packages (from ipywidgets->ax-platform) (3.6.5)\n",
            "Requirement already satisfied: ipython>=4.0.0 in /usr/local/lib/python3.10/dist-packages (from ipywidgets->ax-platform) (7.34.0)\n",
            "Requirement already satisfied: jupyterlab-widgets>=1.0.0 in /usr/local/lib/python3.10/dist-packages (from ipywidgets->ax-platform) (3.0.8)\n",
            "Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/dist-packages (from jinja2->ax-platform) (2.1.3)\n",
            "Requirement already satisfied: python-dateutil>=2.8.1 in /usr/local/lib/python3.10/dist-packages (from pandas->ax-platform) (2.8.2)\n",
            "Requirement already satisfied: pytz>=2020.1 in /usr/local/lib/python3.10/dist-packages (from pandas->ax-platform) (2023.3)\n",
            "Requirement already satisfied: numpy>=1.21.0 in /usr/local/lib/python3.10/dist-packages (from pandas->ax-platform) (1.23.5)\n",
            "Requirement already satisfied: joblib>=1.1.1 in /usr/local/lib/python3.10/dist-packages (from scikit-learn->ax-platform) (1.3.2)\n",
            "Requirement already satisfied: threadpoolctl>=2.0.0 in /usr/local/lib/python3.10/dist-packages (from scikit-learn->ax-platform) (3.2.0)\n",
            "Requirement already satisfied: jupyter-client in /usr/local/lib/python3.10/dist-packages (from ipykernel>=4.5.1->ipywidgets->ax-platform) (6.1.12)\n",
            "Requirement already satisfied: tornado>=4.2 in /usr/local/lib/python3.10/dist-packages (from ipykernel>=4.5.1->ipywidgets->ax-platform) (6.3.1)\n",
            "Requirement already satisfied: setuptools>=18.5 in /usr/local/lib/python3.10/dist-packages (from ipython>=4.0.0->ipywidgets->ax-platform) (67.7.2)\n",
            "Requirement already satisfied: jedi>=0.16 in /usr/local/lib/python3.10/dist-packages (from ipython>=4.0.0->ipywidgets->ax-platform) (0.19.0)\n",
            "Requirement already satisfied: decorator in /usr/local/lib/python3.10/dist-packages (from ipython>=4.0.0->ipywidgets->ax-platform) (4.4.2)\n",
            "Requirement already satisfied: pickleshare in /usr/local/lib/python3.10/dist-packages (from ipython>=4.0.0->ipywidgets->ax-platform) (0.7.5)\n",
            "Requirement already satisfied: prompt-toolkit!=3.0.0,!=3.0.1,<3.1.0,>=2.0.0 in /usr/local/lib/python3.10/dist-packages (from ipython>=4.0.0->ipywidgets->ax-platform) (3.0.39)\n",
            "Requirement already satisfied: pygments in /usr/local/lib/python3.10/dist-packages (from ipython>=4.0.0->ipywidgets->ax-platform) (2.16.1)\n",
            "Requirement already satisfied: backcall in /usr/local/lib/python3.10/dist-packages (from ipython>=4.0.0->ipywidgets->ax-platform) (0.2.0)\n",
            "Requirement already satisfied: matplotlib-inline in /usr/local/lib/python3.10/dist-packages (from ipython>=4.0.0->ipywidgets->ax-platform) (0.1.6)\n",
            "Requirement already satisfied: pexpect>4.3 in /usr/local/lib/python3.10/dist-packages (from ipython>=4.0.0->ipywidgets->ax-platform) (4.8.0)\n",
            "Requirement already satisfied: opt-einsum>=2.3.2 in /usr/local/lib/python3.10/dist-packages (from pyro-ppl>=1.8.4->botorch==0.9.2->ax-platform) (3.3.0)\n",
            "Requirement already satisfied: pyro-api>=0.1.1 in /usr/local/lib/python3.10/dist-packages (from pyro-ppl>=1.8.4->botorch==0.9.2->ax-platform) (0.1.2)\n",
            "Requirement already satisfied: tqdm>=4.36 in /usr/local/lib/python3.10/dist-packages (from pyro-ppl>=1.8.4->botorch==0.9.2->ax-platform) (4.66.0)\n",
            "Requirement already satisfied: six>=1.5 in /usr/local/lib/python3.10/dist-packages (from python-dateutil>=2.8.1->pandas->ax-platform) (1.16.0)\n",
            "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from torch>=1.13.1->botorch==0.9.2->ax-platform) (3.12.2)\n",
            "Requirement already satisfied: typing-extensions in /usr/local/lib/python3.10/dist-packages (from torch>=1.13.1->botorch==0.9.2->ax-platform) (4.7.1)\n",
            "Requirement already satisfied: sympy in /usr/local/lib/python3.10/dist-packages (from torch>=1.13.1->botorch==0.9.2->ax-platform) (1.12)\n",
            "Requirement already satisfied: networkx in /usr/local/lib/python3.10/dist-packages (from torch>=1.13.1->botorch==0.9.2->ax-platform) (3.1)\n",
            "Requirement already satisfied: triton==2.0.0 in /usr/local/lib/python3.10/dist-packages (from torch>=1.13.1->botorch==0.9.2->ax-platform) (2.0.0)\n",
            "Requirement already satisfied: cmake in /usr/local/lib/python3.10/dist-packages (from triton==2.0.0->torch>=1.13.1->botorch==0.9.2->ax-platform) (3.27.1)\n",
            "Requirement already satisfied: lit in /usr/local/lib/python3.10/dist-packages (from triton==2.0.0->torch>=1.13.1->botorch==0.9.2->ax-platform) (16.0.6)\n",
            "Requirement already satisfied: notebook>=4.4.1 in /usr/local/lib/python3.10/dist-packages (from widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (6.4.8)\n",
            "Requirement already satisfied: parso<0.9.0,>=0.8.3 in /usr/local/lib/python3.10/dist-packages (from jedi>=0.16->ipython>=4.0.0->ipywidgets->ax-platform) (0.8.3)\n",
            "Requirement already satisfied: pyzmq>=17 in /usr/local/lib/python3.10/dist-packages (from notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (23.2.1)\n",
            "Requirement already satisfied: argon2-cffi in /usr/local/lib/python3.10/dist-packages (from notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (21.3.0)\n",
            "Requirement already satisfied: jupyter-core>=4.6.1 in /usr/local/lib/python3.10/dist-packages (from notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (5.3.1)\n",
            "Requirement already satisfied: nbformat in /usr/local/lib/python3.10/dist-packages (from notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (5.9.2)\n",
            "Requirement already satisfied: nbconvert in /usr/local/lib/python3.10/dist-packages (from notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (6.5.4)\n",
            "Requirement already satisfied: nest-asyncio>=1.5 in /usr/local/lib/python3.10/dist-packages (from notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (1.5.7)\n",
            "Requirement already satisfied: Send2Trash>=1.8.0 in /usr/local/lib/python3.10/dist-packages (from notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (1.8.2)\n",
            "Requirement already satisfied: terminado>=0.8.3 in /usr/local/lib/python3.10/dist-packages (from notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (0.17.1)\n",
            "Requirement already satisfied: prometheus-client in /usr/local/lib/python3.10/dist-packages (from notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (0.17.1)\n",
            "Requirement already satisfied: ptyprocess>=0.5 in /usr/local/lib/python3.10/dist-packages (from pexpect>4.3->ipython>=4.0.0->ipywidgets->ax-platform) (0.7.0)\n",
            "Requirement already satisfied: wcwidth in /usr/local/lib/python3.10/dist-packages (from prompt-toolkit!=3.0.0,!=3.0.1,<3.1.0,>=2.0.0->ipython>=4.0.0->ipywidgets->ax-platform) (0.2.6)\n",
            "Requirement already satisfied: mpmath>=0.19 in /usr/local/lib/python3.10/dist-packages (from sympy->torch>=1.13.1->botorch==0.9.2->ax-platform) (1.3.0)\n",
            "Requirement already satisfied: platformdirs>=2.5 in /usr/local/lib/python3.10/dist-packages (from jupyter-core>=4.6.1->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (3.10.0)\n",
            "Requirement already satisfied: argon2-cffi-bindings in /usr/local/lib/python3.10/dist-packages (from argon2-cffi->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (21.2.0)\n",
            "Requirement already satisfied: lxml in /usr/local/lib/python3.10/dist-packages (from nbconvert->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (4.9.3)\n",
            "Requirement already satisfied: beautifulsoup4 in /usr/local/lib/python3.10/dist-packages (from nbconvert->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (4.11.2)\n",
            "Requirement already satisfied: bleach in /usr/local/lib/python3.10/dist-packages (from nbconvert->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (6.0.0)\n",
            "Requirement already satisfied: defusedxml in /usr/local/lib/python3.10/dist-packages (from nbconvert->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (0.7.1)\n",
            "Requirement already satisfied: entrypoints>=0.2.2 in /usr/local/lib/python3.10/dist-packages (from nbconvert->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (0.4)\n",
            "Requirement already satisfied: jupyterlab-pygments in /usr/local/lib/python3.10/dist-packages (from nbconvert->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (0.2.2)\n",
            "Requirement already satisfied: mistune<2,>=0.8.1 in /usr/local/lib/python3.10/dist-packages (from nbconvert->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (0.8.4)\n",
            "Requirement already satisfied: nbclient>=0.5.0 in /usr/local/lib/python3.10/dist-packages (from nbconvert->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (0.8.0)\n",
            "Requirement already satisfied: pandocfilters>=1.4.1 in /usr/local/lib/python3.10/dist-packages (from nbconvert->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (1.5.0)\n",
            "Requirement already satisfied: tinycss2 in /usr/local/lib/python3.10/dist-packages (from nbconvert->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (1.2.1)\n",
            "Requirement already satisfied: fastjsonschema in /usr/local/lib/python3.10/dist-packages (from nbformat->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (2.18.0)\n",
            "Requirement already satisfied: jsonschema>=2.6 in /usr/local/lib/python3.10/dist-packages (from nbformat->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (4.19.0)\n",
            "Requirement already satisfied: attrs>=22.2.0 in /usr/local/lib/python3.10/dist-packages (from jsonschema>=2.6->nbformat->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (23.1.0)\n",
            "Requirement already satisfied: jsonschema-specifications>=2023.03.6 in /usr/local/lib/python3.10/dist-packages (from jsonschema>=2.6->nbformat->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (2023.7.1)\n",
            "Requirement already satisfied: referencing>=0.28.4 in /usr/local/lib/python3.10/dist-packages (from jsonschema>=2.6->nbformat->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (0.30.2)\n",
            "Requirement already satisfied: rpds-py>=0.7.1 in /usr/local/lib/python3.10/dist-packages (from jsonschema>=2.6->nbformat->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (0.9.2)\n",
            "Requirement already satisfied: cffi>=1.0.1 in /usr/local/lib/python3.10/dist-packages (from argon2-cffi-bindings->argon2-cffi->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (1.15.1)\n",
            "Requirement already satisfied: soupsieve>1.2 in /usr/local/lib/python3.10/dist-packages (from beautifulsoup4->nbconvert->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (2.4.1)\n",
            "Requirement already satisfied: webencodings in /usr/local/lib/python3.10/dist-packages (from bleach->nbconvert->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (0.5.1)\n",
            "Requirement already satisfied: pycparser in /usr/local/lib/python3.10/dist-packages (from cffi>=1.0.1->argon2-cffi-bindings->argon2-cffi->notebook>=4.4.1->widgetsnbextension~=3.6.0->ipywidgets->ax-platform) (2.21)\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "from pathlib import Path\n",
        "Path(\"tests\").mkdir(exist_ok=True)"
      ],
      "metadata": {
        "id": "4fRuCMT_M3nI"
      },
      "execution_count": 14,
      "outputs": []
    },
    {
      "cell_type": "code",
      "execution_count": 15,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "UBIC59lNLT54",
        "outputId": "2dbe4745-2320-4b01-a100-81d68e90cb71"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Overwriting tests/test_ax.py\n"
          ]
        }
      ],
      "source": [
        "%%writefile tests/test_ax.py\n",
        "def test_simple():\n",
        "  from ax.service.ax_client import AxClient\n",
        "  from ax.utils.measurement.synthetic_functions import branin\n",
        "\n",
        "\n",
        "  ax_client = AxClient()\n",
        "  ax_client.create_experiment(\n",
        "      parameters=[\n",
        "          {\"name\": \"x1\", \"type\": \"range\", \"bounds\": [-5.0, 10.0]},\n",
        "          {\"name\": \"x2\", \"type\": \"range\", \"bounds\": [0.0, 10.0]},\n",
        "      ],\n",
        "      objective_name=\"branin\",\n",
        "      minimize=True,\n",
        "  )\n",
        "\n",
        "  for _ in range(15):\n",
        "      parameters, trial_index = ax_client.get_next_trial()\n",
        "      results = branin(parameters[\"x1\"], parameters[\"x2\"])\n",
        "      ax_client.complete_trial(trial_index=trial_index, raw_data=results)\n",
        "\n",
        "\n",
        "  best_parameters, metrics = ax_client.get_best_parameters()\n"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "import pytest\n",
        "import time\n",
        "\n",
        "from ax.modelbridge.factory import Models # COMMENT, AND IT RUNS FINE\n",
        "\n",
        "class ResultsCollector:\n",
        "    def __init__(self):\n",
        "        self.reports = []\n",
        "        self.collected = 0\n",
        "        self.exitcode = 0\n",
        "        self.passed = 0\n",
        "        self.failed = 0\n",
        "        self.xfailed = 0\n",
        "        self.skipped = 0\n",
        "        self.total_duration = 0\n",
        "\n",
        "    @pytest.hookimpl(hookwrapper=True)\n",
        "    def pytest_runtest_makereport(self, item, call):\n",
        "        outcome = yield\n",
        "        report = outcome.get_result()\n",
        "        if report.when == \"call\":\n",
        "            self.reports.append(report)\n",
        "\n",
        "    def pytest_collection_modifyitems(self, items):\n",
        "        self.collected = len(items)\n",
        "\n",
        "    def pytest_terminal_summary(self, terminalreporter, exitstatus):\n",
        "        print(exitstatus, dir(exitstatus))\n",
        "        self.exitcode = exitstatus\n",
        "        self.passed = terminalreporter.stats.get(\"passed\", [])\n",
        "        self.failed = terminalreporter.stats.get(\"failed\", [])\n",
        "        self.xfailed = terminalreporter.stats.get(\"xfailed\", [])\n",
        "        self.skipped = terminalreporter.stats.get(\"skipped\", [])\n",
        "        self.num_passed = len(self.passed)\n",
        "        self.num_failed = len(self.failed)\n",
        "        self.num_xfailed = len(self.xfailed)\n",
        "        self.num_skipped = len(self.skipped)\n",
        "\n",
        "        self.total_duration = time.time() - terminalreporter._sessionstarttime"
      ],
      "metadata": {
        "id": "83FdD7BuLhB7"
      },
      "execution_count": 16,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "import os\n",
        "collector = ResultsCollector()\n",
        "retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "LeUY6s-nLyzF",
        "outputId": "b9f0f18b-7ce4-44a3-a71d-de5990692b6f"
      },
      "execution_count": 17,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "============================= test session starts ==============================\n",
            "platform linux -- Python 3.10.12, pytest-7.4.0, pluggy-1.2.0 -- /usr/bin/python3\n",
            "cachedir: .pytest_cache\n",
            "rootdir: /content\n",
            "plugins: jaxtyping-0.2.20, typeguard-2.13.3, anyio-3.7.1\n",
            "collecting ... collected 1 item\n",
            "\n",
            "tests/test_ax.py::test_simple PASSED                                     [100%]\n",
            "0 ['__abs__', '__add__', '__and__', '__bool__', '__ceil__', '__class__', '__delattr__', '__dir__', '__divmod__', '__doc__', '__eq__', '__float__', '__floor__', '__floordiv__', '__format__', '__ge__', '__getattribute__', '__getnewargs__', '__gt__', '__hash__', '__index__', '__init__', '__init_subclass__', '__int__', '__invert__', '__le__', '__lshift__', '__lt__', '__mod__', '__mul__', '__ne__', '__neg__', '__new__', '__or__', '__pos__', '__pow__', '__radd__', '__rand__', '__rdivmod__', '__reduce__', '__reduce_ex__', '__repr__', '__rfloordiv__', '__rlshift__', '__rmod__', '__rmul__', '__ror__', '__round__', '__rpow__', '__rrshift__', '__rshift__', '__rsub__', '__rtruediv__', '__rxor__', '__setattr__', '__sizeof__', '__str__', '__sub__', '__subclasshook__', '__truediv__', '__trunc__', '__xor__', 'as_integer_ratio', 'bit_count', 'bit_length', 'conjugate', 'denominator', 'from_bytes', 'imag', 'numerator', 'real', 'to_bytes']\n",
            "\n",
            "============================== 1 passed in 14.77s ==============================\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "report = collector.reports[0]\n",
        "print(report.capstderr)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "muU4NlbCNT-i",
        "outputId": "3f0c16c5-1427-4d4b-a037-49a1867ade3a"
      },
      "execution_count": 18,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 6, in test_simple\n",
            "    ax_client = AxClient()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 203, in __init__\n",
            "    logger.info(\n",
            "Message: 'Starting optimization with verbose logging. To disable logging, set the `verbose_logging` argument to `False`. Note that float values in the logs are rounded to 6 decimal points.'\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 7, in test_simple\n",
            "    ax_client.create_experiment(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 352, in create_experiment\n",
            "    experiment = self.make_experiment(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/utils/instantiation.py\", line 908, in make_experiment\n",
            "    search_space=cls.make_search_space(parameters, parameter_constraints),\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/utils/instantiation.py\", line 687, in make_search_space\n",
            "    typed_parameters = [cls.parameter_from_json(p) for p in parameters]\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/utils/instantiation.py\", line 687, in <listcomp>\n",
            "    typed_parameters = [cls.parameter_from_json(p) for p in parameters]\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/utils/instantiation.py\", line 309, in parameter_from_json\n",
            "    return cls._make_range_param(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/utils/instantiation.py\", line 209, in _make_range_param\n",
            "    parameter_type=cls._to_parameter_type(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/utils/instantiation.py\", line 186, in _to_parameter_type\n",
            "    logger.info(\n",
            "Message: \"Inferred value type of ParameterType.FLOAT for parameter x1. If that is not the expected value type, you can explicity specify 'value_type' ('int', 'float', 'bool' or 'str') in parameter dict.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 7, in test_simple\n",
            "    ax_client.create_experiment(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 352, in create_experiment\n",
            "    experiment = self.make_experiment(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/utils/instantiation.py\", line 908, in make_experiment\n",
            "    search_space=cls.make_search_space(parameters, parameter_constraints),\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/utils/instantiation.py\", line 687, in make_search_space\n",
            "    typed_parameters = [cls.parameter_from_json(p) for p in parameters]\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/utils/instantiation.py\", line 687, in <listcomp>\n",
            "    typed_parameters = [cls.parameter_from_json(p) for p in parameters]\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/utils/instantiation.py\", line 309, in parameter_from_json\n",
            "    return cls._make_range_param(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/utils/instantiation.py\", line 209, in _make_range_param\n",
            "    parameter_type=cls._to_parameter_type(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/utils/instantiation.py\", line 186, in _to_parameter_type\n",
            "    logger.info(\n",
            "Message: \"Inferred value type of ParameterType.FLOAT for parameter x2. If that is not the expected value type, you can explicity specify 'value_type' ('int', 'float', 'bool' or 'str') in parameter dict.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 7, in test_simple\n",
            "    ax_client.create_experiment(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 352, in create_experiment\n",
            "    experiment = self.make_experiment(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/utils/instantiation.py\", line 908, in make_experiment\n",
            "    search_space=cls.make_search_space(parameters, parameter_constraints),\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/utils/instantiation.py\", line 726, in make_search_space\n",
            "    logger.info(f\"Created search space: {ss}.\")\n",
            "Message: \"Created search space: SearchSpace(parameters=[RangeParameter(name='x1', parameter_type=FLOAT, range=[-5.0, 10.0]), RangeParameter(name='x2', parameter_type=FLOAT, range=[0.0, 10.0])], parameter_constraints=[]).\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 7, in test_simple\n",
            "    ax_client.create_experiment(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 373, in create_experiment\n",
            "    self._set_generation_strategy(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 1684, in _set_generation_strategy\n",
            "    self._generation_strategy = choose_generation_strategy(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/modelbridge/dispatch_utils.py\", line 422, in choose_generation_strategy\n",
            "    suggested_model = _suggest_gp_model(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/modelbridge/dispatch_utils.py\", line 242, in _suggest_gp_model\n",
            "    logger.info(f\"Using {method} since {reason}\")\n",
            "Message: 'Using Models.GPEI since there are more ordered parameters than there are categories for the unordered categorical parameters.'\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 7, in test_simple\n",
            "    ax_client.create_experiment(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 373, in create_experiment\n",
            "    self._set_generation_strategy(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 1684, in _set_generation_strategy\n",
            "    self._generation_strategy = choose_generation_strategy(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/modelbridge/dispatch_utils.py\", line 461, in choose_generation_strategy\n",
            "    logger.info(\n",
            "Message: 'Calculating the number of remaining initialization trials based on num_initialization_trials=None max_initialization_trials=None num_tunable_parameters=2 num_trials=None use_batch_trials=False'\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 7, in test_simple\n",
            "    ax_client.create_experiment(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 373, in create_experiment\n",
            "    self._set_generation_strategy(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 1684, in _set_generation_strategy\n",
            "    self._generation_strategy = choose_generation_strategy(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/modelbridge/dispatch_utils.py\", line 479, in choose_generation_strategy\n",
            "    logger.info(\n",
            "Message: 'calculated num_initialization_trials=5'\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 7, in test_simple\n",
            "    ax_client.create_experiment(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 373, in create_experiment\n",
            "    self._set_generation_strategy(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 1684, in _set_generation_strategy\n",
            "    self._generation_strategy = choose_generation_strategy(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/modelbridge/dispatch_utils.py\", line 485, in choose_generation_strategy\n",
            "    logger.info(\n",
            "Message: 'num_completed_initialization_trials=0 num_remaining_initialization_trials=5'\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 7, in test_simple\n",
            "    ax_client.create_experiment(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 373, in create_experiment\n",
            "    self._set_generation_strategy(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 1684, in _set_generation_strategy\n",
            "    self._generation_strategy = choose_generation_strategy(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/modelbridge/dispatch_utils.py\", line 541, in choose_generation_strategy\n",
            "    logger.info(\n",
            "Message: \"Using Bayesian Optimization generation strategy: GenerationStrategy(name='Sobol+GPEI', steps=[Sobol for 5 trials, GPEI for subsequent trials]). Iterations after 5 will take longer to generate due to model-fitting.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 17, in test_simple\n",
            "    parameters, trial_index = ax_client.get_next_trial()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/utils/common/executils.py\", line 161, in actual_wrapper\n",
            "    return func(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 554, in get_next_trial\n",
            "    logger.info(\n",
            "Message: \"Generated new trial 0 with parameters {'x1': 8.16687, 'x2': 5.781874}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 19, in test_simple\n",
            "    ax_client.complete_trial(trial_index=trial_index, raw_data=results)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 772, in complete_trial\n",
            "    logger.info(f\"Completed trial {trial_index} with data: \" f\"{data_update_repr}.\")\n",
            "Message: \"Completed trial 0 with data: {'branin': (24.379399, None)}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 17, in test_simple\n",
            "    parameters, trial_index = ax_client.get_next_trial()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/utils/common/executils.py\", line 161, in actual_wrapper\n",
            "    return func(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 554, in get_next_trial\n",
            "    logger.info(\n",
            "Message: \"Generated new trial 1 with parameters {'x1': 0.840382, 'x2': 3.634383}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 19, in test_simple\n",
            "    ax_client.complete_trial(trial_index=trial_index, raw_data=results)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 772, in complete_trial\n",
            "    logger.info(f\"Completed trial {trial_index} with data: \" f\"{data_update_repr}.\")\n",
            "Message: \"Completed trial 1 with data: {'branin': (17.659245, None)}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 17, in test_simple\n",
            "    parameters, trial_index = ax_client.get_next_trial()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/utils/common/executils.py\", line 161, in actual_wrapper\n",
            "    return func(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 554, in get_next_trial\n",
            "    logger.info(\n",
            "Message: \"Generated new trial 2 with parameters {'x1': 7.793215, 'x2': 8.095407}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 19, in test_simple\n",
            "    ax_client.complete_trial(trial_index=trial_index, raw_data=results)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 772, in complete_trial\n",
            "    logger.info(f\"Completed trial {trial_index} with data: \" f\"{data_update_repr}.\")\n",
            "Message: \"Completed trial 2 with data: {'branin': (54.842551, None)}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 17, in test_simple\n",
            "    parameters, trial_index = ax_client.get_next_trial()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/utils/common/executils.py\", line 161, in actual_wrapper\n",
            "    return func(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 554, in get_next_trial\n",
            "    logger.info(\n",
            "Message: \"Generated new trial 3 with parameters {'x1': 5.883465, 'x2': 3.379981}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 19, in test_simple\n",
            "    ax_client.complete_trial(trial_index=trial_index, raw_data=results)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 772, in complete_trial\n",
            "    logger.info(f\"Completed trial {trial_index} with data: \" f\"{data_update_repr}.\")\n",
            "Message: \"Completed trial 3 with data: {'branin': (24.007451, None)}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 17, in test_simple\n",
            "    parameters, trial_index = ax_client.get_next_trial()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/utils/common/executils.py\", line 161, in actual_wrapper\n",
            "    return func(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 554, in get_next_trial\n",
            "    logger.info(\n",
            "Message: \"Generated new trial 4 with parameters {'x1': 1.623362, 'x2': 3.25036}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 19, in test_simple\n",
            "    ax_client.complete_trial(trial_index=trial_index, raw_data=results)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 772, in complete_trial\n",
            "    logger.info(f\"Completed trial {trial_index} with data: \" f\"{data_update_repr}.\")\n",
            "Message: \"Completed trial 4 with data: {'branin': (9.75195, None)}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 17, in test_simple\n",
            "    parameters, trial_index = ax_client.get_next_trial()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/utils/common/executils.py\", line 161, in actual_wrapper\n",
            "    return func(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 554, in get_next_trial\n",
            "    logger.info(\n",
            "Message: \"Generated new trial 5 with parameters {'x1': 0.147949, 'x2': 0.422576}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 19, in test_simple\n",
            "    ax_client.complete_trial(trial_index=trial_index, raw_data=results)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 772, in complete_trial\n",
            "    logger.info(f\"Completed trial {trial_index} with data: \" f\"{data_update_repr}.\")\n",
            "Message: \"Completed trial 5 with data: {'branin': (48.063925, None)}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 17, in test_simple\n",
            "    parameters, trial_index = ax_client.get_next_trial()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/utils/common/executils.py\", line 161, in actual_wrapper\n",
            "    return func(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 554, in get_next_trial\n",
            "    logger.info(\n",
            "Message: \"Generated new trial 6 with parameters {'x1': 2.936072, 'x2': 4.780238}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 19, in test_simple\n",
            "    ax_client.complete_trial(trial_index=trial_index, raw_data=results)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 772, in complete_trial\n",
            "    logger.info(f\"Completed trial {trial_index} with data: \" f\"{data_update_repr}.\")\n",
            "Message: \"Completed trial 6 with data: {'branin': (6.073245, None)}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 17, in test_simple\n",
            "    parameters, trial_index = ax_client.get_next_trial()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/utils/common/executils.py\", line 161, in actual_wrapper\n",
            "    return func(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 554, in get_next_trial\n",
            "    logger.info(\n",
            "Message: \"Generated new trial 7 with parameters {'x1': -0.900892, 'x2': 5.574406}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 19, in test_simple\n",
            "    ax_client.complete_trial(trial_index=trial_index, raw_data=results)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 772, in complete_trial\n",
            "    logger.info(f\"Completed trial {trial_index} with data: \" f\"{data_update_repr}.\")\n",
            "Message: \"Completed trial 7 with data: {'branin': (19.820355, None)}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 17, in test_simple\n",
            "    parameters, trial_index = ax_client.get_next_trial()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/utils/common/executils.py\", line 161, in actual_wrapper\n",
            "    return func(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 554, in get_next_trial\n",
            "    logger.info(\n",
            "Message: \"Generated new trial 8 with parameters {'x1': -5.0, 'x2': 3.965143}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 19, in test_simple\n",
            "    ax_client.complete_trial(trial_index=trial_index, raw_data=results)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 772, in complete_trial\n",
            "    logger.info(f\"Completed trial {trial_index} with data: \" f\"{data_update_repr}.\")\n",
            "Message: \"Completed trial 8 with data: {'branin': (187.550766, None)}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 17, in test_simple\n",
            "    parameters, trial_index = ax_client.get_next_trial()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/utils/common/executils.py\", line 161, in actual_wrapper\n",
            "    return func(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 554, in get_next_trial\n",
            "    logger.info(\n",
            "Message: \"Generated new trial 9 with parameters {'x1': 1.428244, 'x2': 6.727465}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 19, in test_simple\n",
            "    ax_client.complete_trial(trial_index=trial_index, raw_data=results)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 772, in complete_trial\n",
            "    logger.info(f\"Completed trial {trial_index} with data: \" f\"{data_update_repr}.\")\n",
            "Message: \"Completed trial 9 with data: {'branin': (18.855698, None)}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 17, in test_simple\n",
            "    parameters, trial_index = ax_client.get_next_trial()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/utils/common/executils.py\", line 161, in actual_wrapper\n",
            "    return func(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 554, in get_next_trial\n",
            "    logger.info(\n",
            "Message: \"Generated new trial 10 with parameters {'x1': 1.721007, 'x2': 4.827048}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 19, in test_simple\n",
            "    ax_client.complete_trial(trial_index=trial_index, raw_data=results)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 772, in complete_trial\n",
            "    logger.info(f\"Completed trial {trial_index} with data: \" f\"{data_update_repr}.\")\n",
            "Message: \"Completed trial 10 with data: {'branin': (9.963725, None)}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 17, in test_simple\n",
            "    parameters, trial_index = ax_client.get_next_trial()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/utils/common/executils.py\", line 161, in actual_wrapper\n",
            "    return func(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 554, in get_next_trial\n",
            "    logger.info(\n",
            "Message: \"Generated new trial 11 with parameters {'x1': 2.981277, 'x2': 3.603788}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 19, in test_simple\n",
            "    ax_client.complete_trial(trial_index=trial_index, raw_data=results)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 772, in complete_trial\n",
            "    logger.info(f\"Completed trial {trial_index} with data: \" f\"{data_update_repr}.\")\n",
            "Message: \"Completed trial 11 with data: {'branin': (1.962083, None)}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 17, in test_simple\n",
            "    parameters, trial_index = ax_client.get_next_trial()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/utils/common/executils.py\", line 161, in actual_wrapper\n",
            "    return func(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 554, in get_next_trial\n",
            "    logger.info(\n",
            "Message: \"Generated new trial 12 with parameters {'x1': -2.047605, 'x2': 10.0}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 19, in test_simple\n",
            "    ax_client.complete_trial(trial_index=trial_index, raw_data=results)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 772, in complete_trial\n",
            "    logger.info(f\"Completed trial {trial_index} with data: \" f\"{data_update_repr}.\")\n",
            "Message: \"Completed trial 12 with data: {'branin': (5.632948, None)}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 17, in test_simple\n",
            "    parameters, trial_index = ax_client.get_next_trial()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/utils/common/executils.py\", line 161, in actual_wrapper\n",
            "    return func(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 554, in get_next_trial\n",
            "    logger.info(\n",
            "Message: \"Generated new trial 13 with parameters {'x1': -4.586628, 'x2': 10.0}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 19, in test_simple\n",
            "    ax_client.complete_trial(trial_index=trial_index, raw_data=results)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 772, in complete_trial\n",
            "    logger.info(f\"Completed trial {trial_index} with data: \" f\"{data_update_repr}.\")\n",
            "Message: \"Completed trial 13 with data: {'branin': (45.006166, None)}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 17, in test_simple\n",
            "    parameters, trial_index = ax_client.get_next_trial()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/utils/common/executils.py\", line 161, in actual_wrapper\n",
            "    return func(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 554, in get_next_trial\n",
            "    logger.info(\n",
            "Message: \"Generated new trial 14 with parameters {'x1': 10.0, 'x2': 1.954268}.\"\n",
            "Arguments: ()\n",
            "--- Logging error ---\n",
            "Traceback (most recent call last):\n",
            "  File \"/usr/lib/python3.10/logging/__init__.py\", line 1103, in emit\n",
            "    stream.write(msg + self.terminator)\n",
            "ValueError: I/O operation on closed file.\n",
            "Call stack:\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 196, in _run_module_as_main\n",
            "    return _run_code(code, main_globals, None,\n",
            "  File \"/usr/lib/python3.10/runpy.py\", line 86, in _run_code\n",
            "    exec(code, run_globals)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel_launcher.py\", line 16, in <module>\n",
            "    app.launch_new_instance()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/traitlets/config/application.py\", line 992, in launch_instance\n",
            "    app.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelapp.py\", line 619, in start\n",
            "    self.io_loop.start()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/platform/asyncio.py\", line 195, in start\n",
            "    self.asyncio_loop.run_forever()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 603, in run_forever\n",
            "    self._run_once()\n",
            "  File \"/usr/lib/python3.10/asyncio/base_events.py\", line 1909, in _run_once\n",
            "    handle._run()\n",
            "  File \"/usr/lib/python3.10/asyncio/events.py\", line 80, in _run\n",
            "    self._context.run(self._callback, *self._args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 685, in <lambda>\n",
            "    lambda f: self._run_callback(functools.partial(callback, future))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/ioloop.py\", line 738, in _run_callback\n",
            "    ret = callback()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 825, in inner\n",
            "    self.ctx_run(self.run)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 786, in run\n",
            "    yielded = self.gen.send(value)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 361, in process_one\n",
            "    yield gen.maybe_future(dispatch(*args))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 261, in dispatch_shell\n",
            "    yield gen.maybe_future(handler(stream, idents, msg))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/kernelbase.py\", line 539, in execute_request\n",
            "    self.do_execute(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/tornado/gen.py\", line 234, in wrapper\n",
            "    yielded = ctx_run(next, result)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/ipkernel.py\", line 302, in do_execute\n",
            "    res = shell.run_cell(code, store_history=store_history, silent=silent)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ipykernel/zmqshell.py\", line 539, in run_cell\n",
            "    return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 2975, in run_cell\n",
            "    result = self._run_cell(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3030, in _run_cell\n",
            "    return runner(coro)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/async_helpers.py\", line 78, in _pseudo_sync_runner\n",
            "    coro.send(None)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3257, in run_cell_async\n",
            "    has_raised = await self.run_ast_nodes(code_ast.body, cell_name,\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3473, in run_ast_nodes\n",
            "    if (await self.run_code(code, result,  async_=asy)):\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py\", line 3553, in run_code\n",
            "    exec(code_obj, self.user_global_ns, self.user_ns)\n",
            "  File \"<ipython-input-17-5aec5525e093>\", line 3, in <cell line: 3>\n",
            "    retcode = pytest.main([\"-v\", \"tests/\"], plugins=[collector])\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/config/__init__.py\", line 166, in main\n",
            "    ret: Union[ExitCode, int] = config.hook.pytest_cmdline_main(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 317, in pytest_cmdline_main\n",
            "    return wrap_session(config, _main)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 270, in wrap_session\n",
            "    session.exitstatus = doit(config, session) or 0\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 324, in _main\n",
            "    config.hook.pytest_runtestloop(session=session)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/main.py\", line 349, in pytest_runtestloop\n",
            "    item.config.hook.pytest_runtest_protocol(item=item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 114, in pytest_runtest_protocol\n",
            "    runtestprotocol(item, nextitem=nextitem)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 133, in runtestprotocol\n",
            "    reports.append(call_and_report(item, \"call\", log))\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 222, in call_and_report\n",
            "    call = call_runtest_hook(item, when, **kwds)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 261, in call_runtest_hook\n",
            "    return CallInfo.from_call(\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 341, in from_call\n",
            "    result: Optional[TResult] = func()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 262, in <lambda>\n",
            "    lambda: ihook(item=item, **kwds), when=when, reraise=reraise\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/runner.py\", line 169, in pytest_runtest_call\n",
            "    item.runtest()\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 1788, in runtest\n",
            "    self.ihook.pytest_pyfunc_call(pyfuncitem=self)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_hooks.py\", line 433, in __call__\n",
            "    return self._hookexec(self.name, self._hookimpls, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_manager.py\", line 112, in _hookexec\n",
            "    return self._inner_hookexec(hook_name, methods, kwargs, firstresult)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/pluggy/_callers.py\", line 80, in _multicall\n",
            "    res = hook_impl.function(*args)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/_pytest/python.py\", line 194, in pytest_pyfunc_call\n",
            "    result = testfunction(**testargs)\n",
            "  File \"/content/tests/test_ax.py\", line 19, in test_simple\n",
            "    ax_client.complete_trial(trial_index=trial_index, raw_data=results)\n",
            "  File \"/usr/local/lib/python3.10/dist-packages/ax/service/ax_client.py\", line 772, in complete_trial\n",
            "    logger.info(f\"Completed trial {trial_index} with data: \" f\"{data_update_repr}.\")\n",
            "Message: \"Completed trial 14 with data: {'branin': (3.042888, None)}.\"\n",
            "Arguments: ()\n",
            "\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "print(report.capstdout)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "m8sdIvi1NakP",
        "outputId": "a688e23a-5fd4-4c8b-d445-d5001f6fb35d"
      },
      "execution_count": 11,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "!pytest"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "YZnhaz9KMM5j",
        "outputId": "82969f5a-3618-4770-b1ad-699faff9039f"
      },
      "execution_count": 12,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\u001b[1m============================= test session starts ==============================\u001b[0m\n",
            "platform linux -- Python 3.10.12, pytest-7.4.0, pluggy-1.2.0\n",
            "rootdir: /content\n",
            "plugins: jaxtyping-0.2.20, typeguard-2.13.3, anyio-3.7.1\n",
            "\u001b[1mcollecting ... \u001b[0m\u001b[1m\rcollected 1 item                                                               \u001b[0m\n",
            "\n",
            "tests/test_ax.py \u001b[32m.\u001b[0m\u001b[32m                                                       [100%]\u001b[0m\n",
            "\n",
            "\u001b[32m============================== \u001b[32m\u001b[1m1 passed\u001b[0m\u001b[32m in 13.02s\u001b[0m\u001b[32m ==============================\u001b[0m\n"
          ]
        }
      ]
    }
  ]
}