Unverified Commit 2e23e477 authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Minor refactoring based on static analysis on the code of images: (#2891)

- Replace deprecated string.h with cstring.
- Switch to nullptr.
parent 5cb77a20
#include "read_image_cpu.h"
#include <string.h>
#include <cstring>
torch::Tensor decode_image(const torch::Tensor& data) {
// Check that the input tensor dtype is uint8
......
......@@ -46,7 +46,7 @@ void write_file(std::string filename, torch::Tensor& data) {
auto fileCStr = filename.c_str();
FILE* outfile = fopen(fileCStr, "wb");
TORCH_CHECK(outfile != NULL, "Error opening output file");
TORCH_CHECK(outfile != nullptr, "Error opening output file");
fwrite(fileBytes, sizeof(uint8_t), data.numel(), outfile);
fclose(outfile);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment