correct mistaie in previous commit

[adimux]

In TF tests, verify that TF version is supported, not pytorch's

Description of changes:

Style and formatting:

I have run pre-commit install && pre-commit run --all-files to ensure that auto-formatting happens with every commit.

Issue number, if available

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

[codecov-commenter]

Codecov Report

Merging #528 (69d9866) into master (7d857c9) will increase coverage by 8.37%.
The diff coverage is n/a.

@@            Coverage Diff             @@
##           master     #528      +/-   ##
==========================================
+ Coverage   62.87%   71.24%   +8.37%     
==========================================
  Files         127      117      -10     
  Lines       11117    10591     -526     
==========================================
+ Hits         6990     7546     +556     
+ Misses       4127     3045    -1082     

Impacted Files	Coverage Δ
smdebug/tensorflow/reductions.py	0.00% <0.00%> (-47.06%)	⬇️
smdebug/tensorflow/keras.py	57.42% <0.00%> (-14.46%)	⬇️
...mdebug/profiler/analysis/notebook_utils/heatmap.py	0.00% <0.00%> (-13.01%)	⬇️
smdebug/profiler/utils.py	65.43% <0.00%> (-11.73%)	⬇️
smdebug/xgboost/hook.py	0.00% <0.00%> (-8.43%)	⬇️
smdebug/core/logger.py	70.83% <0.00%> (-5.56%)	⬇️
smdebug/profiler/profiler_config_parser.py	83.51% <0.00%> (-3.73%)	⬇️
smdebug/profiler/analysis/python_stats_reader.py	94.28% <0.00%> (-1.43%)	⬇️
smdebug/profiler/python_profiler.py	83.33% <0.00%> (-0.93%)	⬇️
smdebug/pytorch/hook.py	77.88% <0.00%> (-0.65%)	⬇️
... and 63 more

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 7d857c9...69d9866. Read the comment docs.