fix pre-commit checks

graphcore-research · Oct 10, 2023 · 1079111 · 1079111
1 parent 1299031
commit 1079111
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 7 deletions.
diff --git a/notebooks/binom_factor_table.ipynb b/notebooks/binom_factor_table.ipynb
@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 1,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -35,7 +35,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 2,
    "metadata": {},
    "outputs": [
     {
@@ -60,9 +60,16 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 3,
    "metadata": {},
    "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "WARNING:absl:No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)\n"
+     ]
+    },
     {
      "name": "stdout",
      "output_type": "stream",
@@ -142,7 +149,7 @@
        "a**3 + 9*a**2*b + 9*a*b**2 + b**3"
       ]
      },
-     "execution_count": 7,
+     "execution_count": 3,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -174,7 +181,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 4,
    "metadata": {},
    "outputs": [
     {
@@ -235,7 +242,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 5,
    "metadata": {},
    "outputs": [
     {
@@ -419,7 +426,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 23,
+   "execution_count": 6,
    "metadata": {},
    "outputs": [
     {
@@ -486,7 +493,11 @@
     "inds, weights[inds]\n",
     "\n",
     "with open(\"../pyscf_ipu/experimental/binom_factor_table.py\", \"w\") as f:\n",
+    "  print(\"# Copyright (c) 2023 Graphcore Ltd. All rights reserved.\", file=f)\n",
     "  print(\"# AUTOGENERATED from notebooks/binom_factor_table.ipynb\", file=f)\n",
+    "  print(\"# fmt: off\", file=f)\n",
+    "  print(\"# flake8: noqa\", file=f)\n",
+    "  print(\"# isort: skip_file\", file=f)\n",
     "  print(\"from numpy import array\", file=f)\n",
     "  print(\"binom_factor_table = \", repr((inds, weights[inds])), file=f)\n",
     "\n",

diff --git a/pyscf_ipu/experimental/binom_factor_table.py b/pyscf_ipu/experimental/binom_factor_table.py
@@ -1,4 +1,8 @@
+# Copyright (c) 2023 Graphcore Ltd. All rights reserved.
 # AUTOGENERATED from notebooks/binom_factor_table.ipynb
+# fmt: off
+# flake8: noqa
+# isort: skip_file
 from numpy import array
 binom_factor_table =  ((array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
        1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2,