Error: unsupported status (Running)

[FelixBenning]

Issue Type: Bug

Running pytest results in the following test summary:

=========================== short test summary info ===========================
ERROR tests/test_basics.py
!!!!!!!!!!!!!!!!!!! Interrupted: 1 error during collection !!!!!!!!!!!!!!!!!!!!
======================= 101 warnings, 1 error in 3.57s ========================
Error: Error: unsupported status (Running)

The test code is the following:

import tensorflow as tf
from sgd_g2 import SGD_G2


def test_linear_optimization():
    var1 = tf.Variable(3.0)
    loss = lambda: var1
    optimizer = SGD_G2()
    for _ in range(10):
        optimizer.minimize(loss, [var1])
    print(var1.numpy())

where sgd_g2 is a custom Optimizer. I imagine this might have something to do with tensorflow doing parallel computing and vscode not liking that or something like that. Anyway running pytest in the terminal results in

============================================================================================== 1 passed, 101 warnings in 5.49s =============================================================================================== 

Extension version: 2021.5.926500501
VS Code version: Code 1.57.0 (b4c1bd0a9b03c749ea011b06c6d2676c8091a70c, 2021-06-09T17:18:42.354Z)
OS version: Windows_NT x64 10.0.19042
Restricted Mode: No

System Info

Item	Value
CPUs	Intel(R) Core(TM) i7-6700 CPU @ 3.40GHz (8 x 3408)
GPU Status	2d_canvas: enabled gpu_compositing: enabled multiple_raster_threads: enabled_on oop_rasterization: enabled opengl: enabled_on rasterization: enabled skia_renderer: enabled_on video_decode: enabled vulkan: disabled_off webgl: enabled webgl2: enabled
Load (avg)	undefined
Memory (System)	15.91GB (4.63GB free)
Process Argv	--crash-reporter-id 74736abc-a148-4775-8281-3e328798370c
Screen Reader	no
VM	0%

[karthiknadig]

Dupe of #15736